| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| reduction_ops_common.cc | 87 std::vector<int64> final_shape; local
92 final_shape.push_back(dim);
97 final_shape.push_back(1);
129 result = ctx->builder()->Reshape(finalized, final_shape);
|
| strided_slice_op.cc | 47 TensorShape final_shape; variable
71 shrink_axis_mask_, &dummy_processing_shape, &final_shape,
99 slice = ctx->builder()->Reshape(slice, final_shape.dim_sizes());
127 TensorShape processing_shape, final_shape; variable
153 shrink_axis_mask_, &processing_shape, &final_shape, &dummy,
159 ctx, final_shape == dy_shape,
161 " instead of ", final_shape.DebugString()));
238 TensorShape final_shape; variable
268 shrink_axis_mask_, &dummy_processing_shape, &final_shape,
271 if (final_shape.num_elements() == 0 && rhs_shape.num_elements() == 0)
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| strided_slice_op_test.cc | 109 TensorShape processing_shape, final_shape; local
120 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape,
|
| strided_slice_op.cc | 91 TensorShape processing_shape, final_shape; variable
104 &processing_shape, &final_shape, &is_identity,
112 CHECK(tmp.CopyFrom(input, final_shape));
122 CHECK(tmp.CopyFrom(input.Slice(begin[0], end[0]), final_shape));
128 OP_REQUIRES_OK(context, context->allocate_output(0, final_shape, &result));
136 // final_shape being 2D. This isn't strictly necessary, but I don't
141 final_shape.dims() == 2) {
191 TensorShape processing_shape, final_shape; variable
222 shrink_axis_mask, &processing_shape, &final_shape, &is_identity,
228 context, final_shape == dy_shape
283 TensorShape processing_shape, final_shape; variable
[all...] |
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| single_machine_test.cc | 165 auto final_shape = ops::Shape(root.WithOpName("final_shape"), reshaped); local
170 ops::Equal(root.WithOpName("valid"), final_shape, expected_shape);
175 {final_shape.output});
206 "new_shape", "reshaped", "final_shape", "expected_shape",
|
