Searched refs:Reshape (Results 1 - 25 of 86) sorted by relevance

1234

/external/tensorflow/tensorflow/core/kernels/
H A Dreshape_util.h25 void Reshape(OpKernelContext *context, const Tensor &input_indices_in,
H A Dsparse_reshape_op.cc39 Reshape(context, context->input(0), context->input(1), context->input(2),
H A Dquantized_reshape_op_test.cc43 TEST_F(QuantizedReshapeTest, Reshape) {
/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dreshape_motion_test.cc51 auto c = builder.Reshape(a, {6});
52 auto d = builder.Reshape(b, {6});
H A Dreshape_test.cc104 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 Ddot_operation_test.cc491 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 Dreshape_mover_test.cc56 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 Dimplicit_broadcast_remover_test.cc95 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 Dconst_analysis_test.cc38 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 Dcwise_ops.cc101 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 Ddiag_op.cc69 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 Dgather_op.cc101 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 Dreverse_sequence_op.cc106 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 Dfill_op.cc62 data = ctx->builder()->Reshape(data, {});
H A Dpack_op.cc78 // Reshape the inputs to have an extra dimension of size 1.
80 ctx->builder()->Reshape(values[i], child_shape.dim_sizes());
H A Dstack_ops.cc156 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 Dunpack_op.cc79 // Reshape to drop the 'axis' dimension.
80 auto result = ctx->builder()->Reshape(slice, output_shape.dim_sizes());
H A Ddepthtospace_op.cc118 // 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 Dspacetodepth_op.cc132 // 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 Dreshape_op.cc72 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 Dtensor_array_ops.cc223 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 Dreshape_test.py15 """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 Ddata_flow_grad.cc81 auto original_indices = Reshape(
102 grad_outputs->push_back(Reshape(scope, reconstructed, Shape(scope, data)));
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
H A Dscatter.cc105 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 Dcore_test.py103 keras.layers.Reshape,
109 keras.layers.Reshape,
115 keras.layers.Reshape,
121 keras.layers.Reshape,

Completed in 1880 milliseconds

1234