Home
last modified time | relevance | path

Searched refs:slice_begin (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
Dslice_op_test.py314 def _testGradientSlice(self, input_shape, slice_begin, slice_size): argument
323 slice_t = array_ops.slice(a, slice_begin, slice_size)
334 slices.append(slice(slice_begin[i], slice_begin[i] + slice_size[i]))
339 def _testGradientSliceTape(self, input_shape, slice_begin, slice_size): argument
348 slice_t = array_ops.slice(a, slice_begin, slice_size)
359 slices.append(slice(slice_begin[i], slice_begin[i] + slice_size[i]))
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstrided_slice_op.cc253 absl::InlinedVector<int64, 4> slice_begin, slice_end, slice_strides; in Compile() local
256 slice_begin.push_back(begin[i]); in Compile()
262 slice_begin.push_back(input_shape.dim_size(i) - begin[i] - 1); in Compile()
272 slice = xla::Slice(slice, slice_begin, slice_end, slice_strides); in Compile()
673 absl::InlinedVector<xla::XlaOp, 4> slice_begin; in Compile() local
681 slice_begin.push_back(xla::ConstantR0<int64>(ctx->builder(), begin[i])); in Compile()
686 slice_begin.push_back( in Compile()
698 lhs = xla::DynamicUpdateSlice(lhs, rhs, slice_begin); in Compile()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc3231 NodeDef* slice_begin; in GetSimpleSliceAxis() local
3233 TF_RETURN_IF_ERROR(GetInputNode(node->input(1), &slice_begin)); in GetSimpleSliceAxis()
3235 for (const auto* n : {slice_begin, slice_size}) { in GetSimpleSliceAxis()
3241 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_begin, "value")); in GetSimpleSliceAxis()
3242 if (!slice_begin_t.FromProto(slice_begin->attr().at("value").tensor())) { in GetSimpleSliceAxis()
3338 NodeDef* slice_begin; in GetStridedSliceAxis() local
3341 TF_RETURN_IF_ERROR(GetInputNode(node->input(1), &slice_begin)); in GetStridedSliceAxis()
3345 for (const auto* n : {slice_begin, slice_end, slice_strides}) { in GetStridedSliceAxis()
3353 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_begin, "value")); in GetStridedSliceAxis()
3354 if (!slice_begin_t.FromProto(slice_begin->attr().at("value").tensor())) { in GetStridedSliceAxis()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1808 const Tensor* slice_begin = ic->input_tensor(1); in MaybeUpdateNodeContextOutput() local
1809 valid &= slice_begin != nullptr && slice_begin->NumElements() == 1; in MaybeUpdateNodeContextOutput()
1841 begin = slice_begin->dtype() == DT_INT32 in MaybeUpdateNodeContextOutput()
1842 ? slice_begin->flat<int32>()(0) in MaybeUpdateNodeContextOutput()
1843 : slice_begin->flat<int64>()(0); in MaybeUpdateNodeContextOutput()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_ops_n_z.cc1860 SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, in GetSlicedBoundRanges() argument
1886 slice_begin, slice_end, slice_stride); in GetSlicedBoundRanges()
2028 SmallVectorImpl<int64_t> *slice_begin, SmallVectorImpl<int64_t> *slice_end, in GetSlicedShapeAndBoundRanges() argument
2057 slice_begin, slice_end, slice_stride); in GetSlicedShapeAndBoundRanges()
Dtf_generated_ops.td15405 ::llvm::SmallVectorImpl<int64_t> *slice_begin,
15455 ::llvm::SmallVectorImpl<int64_t> *slice_begin,