| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 113 xla::XlaOp red = xla::SliceInDim(input, /*start_index=*/0, in Compile()
116 xla::XlaOp green = xla::SliceInDim(input, /*start_index=*/1, in Compile()
119 xla::XlaOp blue = xla::SliceInDim(input, /*start_index=*/2, in Compile()
150 xla::XlaOp hue = xla::SliceInDim(input, /*start_index=*/0, in Compile()
153 xla::XlaOp saturation = xla::SliceInDim(input, /*start_index=*/1, in Compile()
156 xla::XlaOp value = xla::SliceInDim(input, /*start_index=*/2, in Compile()
246 xla::XlaOp red = xla::SliceInDim(input, /*start_index=*/0, in Compile()
249 xla::XlaOp green = xla::SliceInDim(input, /*start_index=*/1, in Compile()
252 xla::XlaOp blue = xla::SliceInDim(input, /*start_index=*/2, in Compile()
300 xla::XlaOp red = xla::SliceInDim(input, /*start_index=*/0, in Compile()
[all …]
|
| D | mirror_pad_op.cc | 55 xla::SliceInDim(t_rev, dim_size - excluded_edges - lhs_padding, in DoMirrorPad()
57 auto rhs_pad = xla::SliceInDim(t_rev, excluded_edges, in DoMirrorPad()
|
| D | scan_ops.cc | 115 xla::SliceInDim(output, 1, input_shape.dim_size(axis) + 1, 1, axis); in Compile()
119 xla::SliceInDim(output, 0, input_shape.dim_size(axis), 1, axis); in Compile()
|
| D | resampler_ops.cc | 442 auto weight_x = xla::SliceInDim(ratio, /*start_index=*/0, /*limit_index=*/1, in CalculateGradWarp()
444 auto weight_y = xla::SliceInDim(ratio, /*start_index=*/1, /*limit_index=*/2, in CalculateGradWarp()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 522 LocalOp ComputationBuilder::SliceInDim(const LocalOp& operand, in SliceInDim() function in xla::swig::ComputationBuilder
525 return xla::SliceInDim(operand.op(), start_index, limit_index, stride, dimno); in SliceInDim()
|
| D | local_computation_builder.h | 266 LocalOp SliceInDim(const LocalOp& operand, int64 start_index,
|
| D | local_computation_builder.i | 308 %unignore xla::swig::ComputationBuilder::SliceInDim;
|
| D | xla_client.py | 1337 def SliceInDim(self, operand, start_index, limit_index, stride, dimno): member in ComputationBuilder
1350 return self._client.SliceInDim(operand, start_index, limit_index, stride,
|
| D | xla_client_test.py | 1055 c.SliceInDim(
1062 c.SliceInDim(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 350 XlaOp SliceInDim(const XlaOp& operand, int64 start_index, int64 limit_index,
738 friend XlaOp SliceInDim(const XlaOp& operand, int64 start_index,
1201 XlaOp SliceInDim(const XlaOp& operand, int64 start_index, int64 limit_index,
|
| D | xla_builder.cc | 743 XlaOp XlaBuilder::SliceInDim(const XlaOp& operand, int64 start_index, in SliceInDim() function in xla::XlaBuilder
2183 slices.push_back(SliceInDim(operand, /*start_index=*/i * block_size, in AllToAll()
2875 XlaOp SliceInDim(const XlaOp& operand, int64 start_index, int64 limit_index, in SliceInDim() function
2877 return operand.builder()->SliceInDim(operand, start_index, limit_index, in SliceInDim()
|