Home
last modified time | relevance | path

Searched refs:final_shape (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dstrided_slice_op.cc94 TensorShape processing_shape, final_shape; in Compute() local
107 &processing_shape, &final_shape, &is_identity, in Compute()
115 OP_REQUIRES(context, tmp.CopyFrom(input, final_shape), in Compute()
131 OP_REQUIRES(context, tmp.CopyFrom(slice, final_shape), in Compute()
138 OP_REQUIRES_OK(context, context->allocate_output(0, final_shape, &result)); in Compute()
151 final_shape.dims() == 2 && new_axis_mask == 0) { in Compute()
202 TensorShape processing_shape, final_shape; in Compute() local
233 shrink_axis_mask, &processing_shape, &final_shape, &is_identity, in Compute()
239 context, final_shape == dy_shape, in Compute()
241 " instead of ", final_shape.DebugString())); in Compute()
[all …]
Dstrided_slice_op_test.cc106 TensorShape processing_shape, final_shape; in BM_ValidateStridedSliceOp() local
117 new_axis_mask, shrink_axis_mask, &processing_shape, &final_shape, in BM_ValidateStridedSliceOp()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc58 TensorShape processing_shape, TensorShape final_shape, in EmitDynamicSlice() argument
84 ctx, partial_final_shape.AsTensorShape(&final_shape), in EmitDynamicSlice()
193 slice = xla::Reshape(slice, final_shape.dim_sizes()); in EmitDynamicSlice()
225 TensorShape processing_shape, final_shape; in Compile() local
247 ctx, partial_final_shape.AsTensorShape(&final_shape), in Compile()
285 slice = xla::Reshape(slice, final_shape.dim_sizes()); in Compile()
291 for (int64 i = 0; i < final_shape.dims(); ++i) { in Compile()
350 EmitDynamicSlice(ctx, strides, processing_shape, final_shape, in Compile()
393 PartialTensorShape processing_shape, final_shape; in CompileAsDynamicUpdateSlice() local
404 &processing_shape, &final_shape, &dummy, &dummy, &dummy, in CompileAsDynamicUpdateSlice()
[all …]
Dreduction_ops_common.cc92 std::vector<int64> final_shape; in Compile() local
97 final_shape.push_back(dim); in Compile()
102 final_shape.push_back(1); in Compile()
126 auto result = keep_dims_ ? xla::Reshape(finalized, final_shape) : finalized; in Compile()
/external/tensorflow/tensorflow/core/util/
Dstrided_slice_op.cc174 PartialTensorShape* processing_shape, PartialTensorShape* final_shape, in ValidateStridedSliceOp() argument
381 final_shape->Clear(); in ValidateStridedSliceOp()
400 final_shape->AddDim(processing_shape->dim_size(gather_index)); in ValidateStridedSliceOp()
406 final_shape->AddDim(1); in ValidateStridedSliceOp()
422 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice, in ValidateStridedSliceOp() argument
436 !partial_final_shape.AsTensorShape(final_shape)) { in ValidateStridedSliceOp()
Dstrided_slice_op.h69 PartialTensorShape* processing_shape, PartialTensorShape* final_shape,
81 TensorShape* final_shape, bool* is_identity, bool* is_simple_slice,
/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.h429 ShapeArray final_shape(tensors[0].shape().begin(), tensors[0].shape().end()); in Concat()
430 final_shape[primary_dim] = 0; // We'll build this up as we go along. in Concat()
445 DCHECK_EQ(final_shape[cdim], st_shape[cdim]) in Concat()
449 << ". Expecting shape like: [" << str_util::Join(final_shape, ",") in Concat()
454 final_shape[primary_dim] = in Concat()
455 (final_shape[primary_dim] + st_shape[primary_dim]); in Concat()
462 final_order = UndefinedOrder(final_shape); in Concat()
491 return SparseTensor(output_ix, output_vals, final_shape, final_order); in Concat()
/external/tensorflow/tensorflow/lite/python/
Dlite_v2_test_util.py53 for idx, (shape_signature, final_shape) in enumerate(input_shapes):
57 interpreter.resize_tensor_input(index, final_shape, strict=True)
/external/tensorflow/tensorflow/python/ops/distributions/
Ddirichlet_multinomial.py276 final_shape = array_ops.concat([[n], self.batch_shape_tensor(), [k]], 0)
277 x = array_ops.reshape(x, final_shape)
Dmultinomial.py275 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]
Dutil.py1333 final_shape = head.concatenate(middle.concatenate(tail))
1335 final_shape = None
1338 final_shape = None
1349 if final_shape is not None:
1350 x.set_shape(final_shape)
Dtransformed_distribution.py413 final_shape = array_ops.concat([sample_shape, batch_event_shape], 0)
414 x = array_ops.reshape(x, final_shape)
Ddistribution.py747 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/
Dlinear_operator_circulant.py222 final_shape = vec_leading_shape.concatenate(self.block_shape)
225 final_shape = array_ops.concat(
227 return array_ops.reshape(vec, final_shape)
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine_test.cc178 auto final_shape = ops::Shape(root.WithOpName("final_shape"), reshaped); in TEST_F() local
183 ops::Equal(root.WithOpName("valid"), final_shape, expected_shape); in TEST_F()
188 {final_shape.output}); in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc303 SmallVector<int64_t, 4> final_shape = { in HandleConv2DFilter() local
307 GetReshapeOpForConv2DFilter(final_shape, transpose_op, &builder); in HandleConv2DFilter()
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py5185 final_shape = indices.get_shape()[:batch_dims].merge_with(
5187 final_shape = final_shape.concatenate(indices.get_shape().dims[batch_dims:])
5188 final_shape = final_shape.concatenate(params.get_shape()[batch_dims + 1:])
5189 result.set_shape(final_shape)
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2849 const nvinfer1::Dims* final_shape = nullptr, in ConvertStridedSliceHelper() argument
2894 if (final_shape) { in ConvertStridedSliceHelper()
2896 params->converter, TRT_TensorOrWeights(tensor), *final_shape, in ConvertStridedSliceHelper()
3022 if (final_shape) { in ConvertStridedSliceHelper()
3024 params->converter, TRT_TensorOrWeights(tensor), *final_shape, in ConvertStridedSliceHelper()
3142 PartialTensorShape final_shape; in ConvertStridedSlice() local
3153 &final_shape, &is_identity, &is_simple_slice, &slice_dim0, &begin, &end, in ConvertStridedSlice()
3205 TensorShapeToTrtDims(final_shape, /*ignore_first_dim=*/true); in ConvertStridedSlice()
/external/tensorflow/tensorflow/core/ops/
Darray_ops.cc1740 PartialTensorShape processing_shape, final_shape; in __anondb9326b22502() local
1746 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in __anondb9326b22502()
1750 TF_RETURN_IF_ERROR(c->MakeShapeFromPartialTensorShape(final_shape, &out)); in __anondb9326b22502()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc3367 TensorShape final_shape; in GetStridedSliceAxis() local
3377 &processing_shape, &final_shape, &is_identity, &is_simple_slice, in GetStridedSliceAxis()
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py3935 final_shape = array_ops.concat(
3937 return wrap(array_ops.reshape(values, final_shape), True)