| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 48 TensorShape final_shape; in Compile() local
72 shrink_axis_mask_, &dummy_processing_shape, &final_shape, in Compile()
101 slice = xla::Reshape(slice, final_shape.dim_sizes()); in Compile()
129 TensorShape processing_shape, final_shape; in Compile() local
155 shrink_axis_mask_, &processing_shape, &final_shape, &dummy, in Compile()
161 ctx, final_shape == dy_shape, in Compile()
163 " instead of ", final_shape.DebugString())); in Compile()
240 TensorShape final_shape; in Compile() local
270 shrink_axis_mask_, &dummy_processing_shape, &final_shape, in Compile()
273 if (final_shape.num_elements() == 0 && rhs_shape.num_elements() == 0) { in Compile()
[all …]
|
| D | reduction_ops_common.cc | 87 std::vector<int64> final_shape; in Compile() local
92 final_shape.push_back(dim); in Compile()
97 final_shape.push_back(1); in Compile()
121 auto result = keep_dims_ ? xla::Reshape(finalized, final_shape) : finalized; in Compile()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | strided_slice_op.cc | 92 TensorShape processing_shape, final_shape; in Compute() local
105 &processing_shape, &final_shape, &is_identity, in Compute()
113 OP_REQUIRES(context, tmp.CopyFrom(input, final_shape), in Compute()
129 OP_REQUIRES(context, tmp.CopyFrom(slice, final_shape), in Compute()
136 OP_REQUIRES_OK(context, context->allocate_output(0, final_shape, &result)); in Compute()
149 final_shape.dims() == 2 && new_axis_mask == 0) { in Compute()
199 TensorShape processing_shape, final_shape; in Compute() local
230 shrink_axis_mask, &processing_shape, &final_shape, &is_identity, in Compute()
236 context, final_shape == dy_shape, in Compute()
238 " instead of ", final_shape.DebugString())); in Compute()
[all …]
|
| D | strided_slice_op_test.cc | 109 TensorShape processing_shape, final_shape; in BM_ValidateStridedSliceOp() local
120 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape, in BM_ValidateStridedSliceOp()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | strided_slice_op.cc | 157 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, in ValidateStridedSliceOp() argument
364 final_shape->Clear(); in ValidateStridedSliceOp()
367 final_shape->AddDim(processing_shape->dim_size(gather_index)); in ValidateStridedSliceOp()
369 final_shape->AddDim(1); in ValidateStridedSliceOp()
380 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice, in ValidateStridedSliceOp() argument
393 !partial_final_shape.AsTensorShape(final_shape)) { in ValidateStridedSliceOp()
|
| D | strided_slice_op.h | 48 PartialTensorShape* processing_shape, PartialTensorShape* final_shape,
59 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice,
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 508 ShapeArray final_shape(tensors[0].shape().begin(), tensors[0].shape().end()); in Concat()
509 final_shape[primary_dim] = 0; // We'll build this up as we go along. in Concat()
524 DCHECK_EQ(final_shape[cdim], st_shape[cdim]) in Concat()
528 << ". Expecting shape like: [" << str_util::Join(final_shape, ",") in Concat()
533 final_shape[primary_dim] = in Concat()
534 (final_shape[primary_dim] + st_shape[primary_dim]); in Concat()
541 final_order = UndefinedOrder(final_shape); in Concat()
568 return SparseTensor(output_ix, output_vals, final_shape, final_order); in Concat()
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | multinomial.py | 275 final_shape = array_ops.concat([[n], self.batch_shape_tensor(), [k]], 0)
276 x = array_ops.reshape(x, final_shape) # [n, B1, B2,..., Bm, k]
|
| D | dirichlet_multinomial.py | 276 final_shape = array_ops.concat([[n], self.batch_shape_tensor(), [k]], 0)
277 x = array_ops.reshape(x, final_shape)
|
| D | util.py | 1325 final_shape = head.concatenate(middle.concatenate(tail))
1327 final_shape = None
1330 final_shape = None
1341 if final_shape is not None:
1342 x.set_shape(final_shape)
|
| D | transformed_distribution.py | 413 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
414 x = array_ops.reshape(x, final_shape)
|
| D | distribution.py | 747 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
748 samples = array_ops.reshape(samples, final_shape)
|
| /external/tensorflow/tensorflow/python/ops/linalg/ |
| D | linear_operator_circulant.py | 223 final_shape = vec_leading_shape.concatenate(self.block_shape)
226 final_shape = array_ops.concat(
228 return array_ops.reshape(vec, final_shape)
|
| /external/tensorflow/tensorflow/contrib/distributions/python/ops/ |
| D | conditional_transformed_distribution.py | 89 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
90 x = array_ops.reshape(x, final_shape)
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine_test.cc | 165 auto final_shape = ops::Shape(root.WithOpName("final_shape"), reshaped); in TEST_F() local
170 ops::Equal(root.WithOpName("valid"), final_shape, expected_shape); in TEST_F()
175 {final_shape.output}); in TEST_F()
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| D | beam_search_decoder.py | 177 final_shape = array_ops.shape(gather_from)
181 ordered = array_ops.reshape(ordered, final_shape)
1377 final_shape = array_ops.shape(gather_from)[:1 + len(gather_shape)]
1382 output = array_ops.reshape(output, final_shape, name="output")
|
| D | sampler.py | 713 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
714 return array_ops.reshape(samples, final_shape)
|
| D | helper.py | 72 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
73 return array_ops.reshape(samples, final_shape)
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_ops.py | 3519 final_shape = indices.get_shape()[:batch_dims].merge_with(
3521 final_shape = final_shape.concatenate(indices.get_shape().dims[batch_dims:])
3522 final_shape = final_shape.concatenate(params.get_shape()[batch_dims + 1:])
3523 result.set_shape(final_shape)
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | array_ops.cc | 1601 PartialTensorShape processing_shape, final_shape; in __anon7c94107b2502() local
1607 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in __anon7c94107b2502()
1611 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(final_shape, &out)); in __anon7c94107b2502()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3183 TensorShape final_shape; in GetSliceAxis() local
3193 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in GetSliceAxis()
|