| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_debug.cc | 83 TFE_TensorDebugInfo* debug_info, int dim_index) { in TFE_TensorDebugInfoOnDeviceDim()
|
| D | c_api.cc | 310 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, in TFE_TensorHandleDim()
512 Status Dim(int dim_index, int64* dim) const override { in Dim()
531 int64_t (*dim_callback)(void* data, int dim_index, void* arg, in TFE_NewCustomDeviceTensorHandle()
551 [dim_callback, data, arg](int dim_index, TF_Status* status) { in TFE_NewCustomDeviceTensorHandle()
589 [dimvec](int dim_index, TF_Status* status) { in TFE_NewTensorHandleFromDeviceMemory()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_data.cc | 58 Status LocalTensorHandleData::Dim(int dim_index, int64* dim) const { in Dim()
|
| D | custom_device_test.cc | 76 Status Dim(int dim_index, int64* dim) const override { in Dim()
|
| D | tensor_handle.cc | 91 Status TensorHandle::PackedTensorHandleData::Dim(int dim_index, in Dim()
638 Status TensorHandle::Dim(int dim_index, int64* dim) const { in Dim()
|
| /external/skia/modules/skottie/src/effects/ |
| D | GaussianBlurEffect.cpp | 57 const auto dim_index = SkTPin<size_t>(static_cast<size_t>(fDimensions), in onSync() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops_common.cc | 111 int dim_index = 0; in Simplify() local
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | eager_operation_jni.cc | 135 jint dim_index) { in Java_org_tensorflow_EagerOperation_dim()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_tensor_handle_data.cc | 142 Status RemoteTensorHandleData::Dim(int dim_index, int64* dim) const { in Dim()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder.cc | 759 for (int64 dim_index = 0; dim_index < input_window->dimensions_size(); in RewriteInputWithDynamicPadding() local
1031 for (int64 dim_index = 0; dim_index < rank; ++dim_index) { in RewriteDynamicReduceWindowSamePadding() local
1076 for (int64 dim_index = 0; dim_index < rank; ++dim_index) { in RewriteDynamicSelectAndScatterSamePadding() local
1123 for (int64 dim_index = 0; dim_index < rank; ++dim_index) { in RewriteDynamicSelectAndScatterSamePadding() local
|
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper.cc | 190 for (size_t dim_index = 0; dim_index < PyList_Size(shape); ++dim_index) { in Prepare() local
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api.cc | 192 int32_t TfLiteTensorDim(const TfLiteTensor* tensor, int32_t dim_index) { in TfLiteTensorDim()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 1017 void PrintOneDim(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDim()
1055 void PrintDimSpacing(int dim_index, int num_dims, string* result) { in PrintDimSpacing()
1070 void PrintOneDimV2(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDimV2()
|
| D | shape_inference.cc | 658 int64 dim_index = dim_index_in; in ReplaceDim() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 233 int32 LiteralBase::GetDynamicSize(int64 dim_index, in GetDynamicSize()
462 void LiteralBase::Piece::SetDynamicSize(int64 dim_index, int32 size) { in SetDynamicSize()
528 void MutableLiteralBase::SetDynamicSize(int64 dim_index, int32 size) { in SetDynamicSize()
532 void MutableLiteralBase::SetDynamicSize(int64 dim_index, in SetDynamicSize()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device.cc | 202 int64_t ParallelTensorDim(void* data, int dim_index, void* arg, in ParallelTensorDim()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.cc | 153 int64_t TF_Dim(const TF_Tensor* t, int dim_index) { in TF_Dim()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 132 int64 dim_index; in Compile() local
|
| /external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/ |
| D | Interpreter.cs | 263 private static extern int TfLiteTensorDim(TfLiteTensor tensor, int dim_index); in TfLiteTensorDim()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_legalize_to_parallel_loops.cc | 79 size_t dim_index, int64_t dim, OpBuilder* b) { in GetStaticOrDynamicDim()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 53 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index): argument
|
| D | nn_ops.py | 3602 def _swap_axis(input_tensor, dim_index, last_index, name=None): argument
3945 def _move_dim_to_end(tensor, dim_index, rank): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_n_z.cc | 1399 LogicalResult VerifySplitInputAndSplitDim(Op op, Optional<int64_t> *dim_index) { in VerifySplitInputAndSplitDim()
1435 Optional<int64_t> dim_index; in Verify() local
1465 Optional<int64_t> dim_index = 0; in Verify() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 2153 const int64 dim_index = i - 1; in HandleSliceToDynamic() local
2218 const int64 dim_index = i - 1; in HandlePadToStatic() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 858 const int64 dim_index = i - 1; in EmitPadToStaticFromMlir() local
876 const int64 dim_index = i - 1; in EmitPadToStaticFromMlir() local
996 const int64 dim_index = i - 1; in EmitSliceToDynamicFromMlir() local
|