/external/tensorflow/tensorflow/core/kernels/ |
H A D | reshape_util.h | 25 void Reshape(OpKernelContext *context, const Tensor &input_indices_in,
|
H A D | sparse_reshape_op.cc | 39 Reshape(context, context->input(0), context->input(1), context->input(2),
|
H A D | quantized_reshape_op_test.cc | 43 TEST_F(QuantizedReshapeTest, Reshape) {
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | reshape_motion_test.cc | 51 auto c = builder.Reshape(a, {6}); 52 auto d = builder.Reshape(b, {6});
|
H A D | reshape_test.cc | 104 auto reshape = builder.Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}, 122 builder.Reshape(/*operand=*/a, /*dimensions=*/{}, /*new_sizes=*/{1}); 215 builder.Reshape(/*operand=*/parameter, /*dimensions=*/{0}, 230 builder.Reshape(/*operand=*/parameter, /*dimensions=*/{0}, 249 builder.Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}, 264 builder.Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}, 281 builder.Reshape(/*operand=*/parameter, /*dimensions=*/{1, 0}, 335 builder.Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1}, 351 builder.Reshape(/*operand=*/parameter, /*dimensions=*/{0, 1, 2, 3}, 367 builder.Reshape(/*operan [all...] |
H A D | dot_operation_test.cc | 491 auto x_flat = builder.Reshape(x, {0, 1, 2, 3}, {4, 2, 2}); 492 auto y_flat = builder.Reshape(y, {0, 1, 2, 3}, {4, 2, 2}); 499 x_slice = builder.Reshape(x_slice, {0, 1, 2}, {2, 2}); 501 y_slice = builder.Reshape(y_slice, {0, 1, 2}, {2, 2}); 504 out = builder.Reshape(out, {0, 1}, {1, 2, 2}); 508 builder.Reshape(out_flat, {0, 1, 2}, {2, 2, 2, 2});
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | reshape_mover_test.cc | 56 op::Add(op::Reshape(param0), op::Reshape(param1))); 61 op::Add(op::Reshape(param0), op::Reshape(param1))); 94 op::Add(op::Reshape(rng0), const1)); 99 op::Add(op::Reshape(rng0), const1)); 119 op::Add(op::Reshape(param0), op::Reshape(param1))); 125 op::Add(op::Reshape(op::Parameter()), op::Reshape(o [all...] |
H A D | implicit_broadcast_remover_test.cc | 95 op::Broadcast(op::Reshape(op::Parameter())))); 140 EXPECT_THAT(root, op::Clamp(op::Broadcast(op::Reshape(op::Parameter())), 141 op::Broadcast(op::Reshape(op::Parameter())), 142 op::Broadcast(op::Reshape(op::Parameter())))); 168 op::Broadcast(op::Reshape(op::Parameter())),
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
H A D | const_analysis_test.cc | 38 auto c = ops::Reshape(root, arg2, b); 48 // Arg 1 must be constant because it flows to the shape argument of a Reshape. 49 // Arg 2 is used only as the value input to a Reshape and need not be const. 63 auto a = ops::Reshape(root, arg0, arg1); 64 auto b = ops::Reshape(root, arg2, a);
|
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
H A D | cwise_ops.cc | 101 auto lhs_shaped = builder->Reshape(lhs, broadcast_helper.x_reshape()); 102 auto rhs_shaped = builder->Reshape(rhs, broadcast_helper.y_reshape()); 117 // broadcast dimensions were prepended to the shape. Reshape 127 auto lhs_output = builder->Reshape(lhs_broadcast, lhs_reorder, 134 auto rhs_output = builder->Reshape(rhs_broadcast, rhs_reorder,
|
H A D | diag_op.cc | 69 builder->Reshape(input, broadcast_dims); 109 input = builder->Reshape(input, {size}); 121 diag = builder->Reshape(diag, new_dims); 164 // the Pad -> Reshape -> Slice. 175 diag = builder->Reshape(diag, {size}); 185 diag = builder->Reshape(diag, {new_size, new_size + 1}); 189 diag = builder->Reshape(diag, new_dims); 253 // the Pad -> Reshape -> Slice. 270 diag = builder->Reshape(diag, flattened_dims); 290 // Reshape s [all...] |
H A D | gather_op.cc | 101 auto flat_indices = builder->Reshape(indices, flat_indices_shape); 122 auto indices_offset = bodyb->Reshape(i, {1}); 138 auto slice_i = bodyb->Reshape( 144 loop_out_shape.dims(), bodyb->Reshape(zero_index, {1})); 145 out_index_vals[input_shape_pre_axis.dims()] = bodyb->Reshape(i, {1}); 160 *gather_output = builder->Reshape(outputs[2], out_shape.dim_sizes());
|
H A D | reverse_sequence_op.cc | 106 seq_lens, body_builder->Reshape(i, {1}), {1}); 113 indices, body_builder->Reshape(i, {1}), 114 body_builder->Reshape( 126 body_builder->Reshape( 163 auto mask = builder->Lt(iota, builder->Reshape(seq_lens, dims), {seq_dim_});
|
H A D | fill_op.cc | 62 data = ctx->builder()->Reshape(data, {});
|
H A D | pack_op.cc | 78 // Reshape the inputs to have an extra dimension of size 1. 80 ctx->builder()->Reshape(values[i], child_shape.dim_sizes());
|
H A D | stack_ops.cc | 156 b->Pad(b->Reshape(index, {1}), b->ConstantR0<int32>(0), 161 auto update = b->Reshape(value, slice_shape.dim_sizes()); 211 b->Pad(b->Reshape(index, {1}), b->ConstantR0<int32>(0), 224 ctx->SetOutput(0, b->Reshape(read, value_shape));
|
H A D | unpack_op.cc | 79 // Reshape to drop the 'axis' dimension. 80 auto result = ctx->builder()->Reshape(slice, output_shape.dim_sizes());
|
H A D | depthtospace_op.cc | 118 // 1. Reshape `input` to `reshaped` of shape: 133 xla::ComputationDataHandle reshaped = b->Reshape(input, reshaped_shape); 147 // 3. Reshape `permuted_reshaped` to flatten `block_shape` into the 156 b->Reshape(permuted_reshaped, output_shape);
|
H A D | spacetodepth_op.cc | 132 // 1. Reshape `input` to `reshaped` of shape: 138 xla::ComputationDataHandle reshaped = b->Reshape(input, reshaped_shape); 151 // 3. Reshape `permuted_reshaped` to flatten `block_shape` into the 160 b->Reshape(permuted_reshaped, output_shape);
|
H A D | reshape_op.cc | 72 errors::InvalidArgument("Reshape cannot infer the missing input size " 90 VLOG(1) << "Reshape " << input_shape.DebugString() << " " 94 ctx->builder()->Reshape(ctx->Input(0), shape.dim_sizes())); 98 REGISTER_XLA_OP(Name("Reshape").CompileTimeConstInput("shape"), ReshapeOp);
|
H A D | tensor_array_ops.cc | 223 b->Pad(b->Reshape(index, {1}), b->ConstantR0<int32>(0), 228 auto update = b->Reshape(value, slice_shape.dim_sizes()); 267 b->Pad(b->Reshape(index, {1}), b->ConstantR0<int32>(0), 278 ctx->SetOutput(0, b->Reshape(read, value_shape)); 421 b->Pad(b->Reshape(index, {1}), b->ConstantR0<int32>(0), 461 ctx->SetOutput(0, b->Reshape(ta, shape)); 532 ta, b->Reshape(value, ta_shape.dim_sizes()))));
|
/external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/ |
H A D | reshape_test.py | 15 """Tests for Reshape Bijector.""" 23 from tensorflow.contrib.distributions.python.ops.bijectors.reshape import Reshape namespace 58 bijector = Reshape( 84 bijector = Reshape( 110 bijector = Reshape( 132 bijector = Reshape( 148 bijector = Reshape( 163 bijector = Reshape( 181 bijector = Reshape( 199 bijector = Reshape( [all...] |
/external/tensorflow/tensorflow/cc/gradients/ |
H A D | data_flow_grad.cc | 81 auto original_indices = Reshape( 102 grad_outputs->push_back(Reshape(scope, reconstructed, Shape(scope, data)));
|
/external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
H A D | scatter.cc | 105 auto flat_indices = builder->Reshape(indices, flat_indices_shape); 106 auto flat_updates = builder->Reshape(updates, flat_updates_shape); 126 auto indices_offset = body_builder->Reshape(i, {1}); 151 auto updates_offset = body_builder->Reshape(i, {1}); 168 update = body_builder->Reshape(update, updates_slice_shape);
|
/external/tensorflow/tensorflow/python/keras/_impl/keras/layers/ |
H A D | core_test.py | 103 keras.layers.Reshape, 109 keras.layers.Reshape, 115 keras.layers.Reshape, 121 keras.layers.Reshape,
|