Searched refs:slice_begin (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | strided_slice_op.cc | 76 absl::InlinedVector<int64, 4> slice_begin, slice_end, slice_strides; in Compile() local
80 slice_begin.push_back(begin[i]); in Compile()
86 slice_begin.push_back(input_shape.dim_size(i) - begin[i] - 1); in Compile()
99 slice = xla::Slice(slice, slice_begin, slice_end, slice_strides); in Compile()
291 absl::InlinedVector<xla::XlaOp, 4> slice_begin; in Compile() local
299 slice_begin.push_back(xla::ConstantR0<int64>(ctx->builder(), begin[i])); in Compile()
304 slice_begin.push_back( in Compile()
316 lhs = xla::DynamicUpdateSlice(lhs, rhs, slice_begin); in Compile()
|
| /external/tensorflow/tensorflow/lite/tools/accuracy/ilsvrc/ |
| D | inception_preprocessing.cc | 42 auto slice_begin = ops::Concat(s, {bbox_begin, Input({0})}, 0); in CentralCropImage() local
44 *cropped_image = ops::Slice(s, decoded_image, slice_begin, slice_size); in CentralCropImage()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | slice_op_test.py | 277 def _testGradientSlice(self, input_shape, slice_begin, slice_size): argument
286 slice_t = array_ops.slice(a, slice_begin, slice_size)
297 slices.append(slice(slice_begin[i], slice_begin[i] + slice_size[i]))
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1508 const Tensor* slice_begin = ic->input_tensor(1); in MaybeUpdateNodeContextOutput() local
1509 valid &= slice_begin != nullptr && slice_begin->NumElements() == 1; in MaybeUpdateNodeContextOutput()
1541 begin = slice_begin->dtype() == DT_INT32 in MaybeUpdateNodeContextOutput()
1542 ? slice_begin->flat<int32>()(0) in MaybeUpdateNodeContextOutput()
1543 : slice_begin->flat<int64>()(0); in MaybeUpdateNodeContextOutput()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 3154 NodeDef* slice_begin; in GetSliceAxis() local
3157 TF_RETURN_IF_ERROR(GetInputNode(node->input(1), &slice_begin)); in GetSliceAxis()
3161 for (const auto* n : {slice_begin, slice_end, slice_strides}) { in GetSliceAxis()
3169 TF_RETURN_IF_ERROR(CheckAttrExists(*slice_begin, "value")); in GetSliceAxis()
3170 if (!slice_begin_t.FromProto(slice_begin->attr().at("value").tensor())) { in GetSliceAxis()
|