Searched refs:slice_stride (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | slice_test.cc | 207 int64 slice_stride; member
226 {spec.slice_stride}); in Run()
231 i += spec.slice_stride) { in Run()
247 spec.slice_limit, spec.slice_stride); in SliceR1TestDataToString()
|
| /external/mesa3d/src/gallium/auxiliary/driver_trace/ |
| D | tr_dump.c | 449 unsigned slice_stride) in trace_dump_box_bytes() argument
459 + (box->depth - 1) * slice_stride; in trace_dump_box_bytes()
|
| D | tr_dump.h | 94 unsigned slice_stride);
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | tensorflow_op_layer_test.py | 588 slice_stride = keras.Input(shape=(), dtype='int32')
590 out = inp[..., first_dim[0], slice_start[0]:slice_stop[0]:slice_stride[0]]
592 inputs=[inp, first_dim, slice_start, slice_stop, slice_stride],
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 1812 const Tensor* slice_stride = ic->input_tensor(3); in MaybeUpdateNodeContextOutput() local
1813 valid &= slice_stride != nullptr && slice_stride->NumElements() == 1; in MaybeUpdateNodeContextOutput()
1851 int64 stride = slice_stride->dtype() == DT_INT32 in MaybeUpdateNodeContextOutput()
1852 ? slice_stride->flat<int32>()(0) in MaybeUpdateNodeContextOutput()
1853 : slice_stride->flat<int64>()(0); in MaybeUpdateNodeContextOutput()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1861 SmallVectorImpl<int64_t> *slice_stride) { in GetSlicedBoundRanges() argument
1886 slice_begin, slice_end, slice_stride); in GetSlicedBoundRanges()
2029 SmallVectorImpl<int64_t> *slice_stride) { in GetSlicedShapeAndBoundRanges() argument
2057 slice_begin, slice_end, slice_stride); in GetSlicedShapeAndBoundRanges()
|
| D | tf_generated_ops.td | 15407 ::llvm::SmallVectorImpl<int64_t> *slice_stride);
15457 ::llvm::SmallVectorImpl<int64_t> *slice_stride);
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | hlo_ops.cc | 227 llvm::SmallVector<int64_t, 8> slice_stride(slice_end.size(), 1); in matchAndRewrite() local
238 GetI64ElementsAttr(slice_stride, &rewriter)); in matchAndRewrite()
|