| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_debug.cc | 82 TFE_TensorDebugInfo* debug_info, int dim_index) { in TFE_TensorDebugInfoOnDeviceDim()
|
| D | custom_device_testutil.cc | 48 int64_t LoggedTensorDim(void* data, int dim_index, TF_Status* status) { in LoggedTensorDim()
|
| D | c_api.cc | 315 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, in TFE_TensorHandleDim()
509 Status Dim(int dim_index, int64_t* dim) const override { in Dim()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_data.cc | 58 Status LocalTensorHandleData::Dim(int dim_index, int64_t* dim) const { in Dim()
|
| D | custom_device_test.cc | 79 Status Dim(int dim_index, int64_t* 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_t* 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/lite/c/ |
| D | c_api_opaque.cc | 30 int32_t dim_index) { in TfLiteOpaqueTensorDim()
|
| D | c_api.cc | 208 int32_t TfLiteTensorDim(const TfLiteTensor* tensor, int32_t dim_index) { in TfLiteTensorDim()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops_common.cc | 111 int dim_index = 0; in Simplify() local
|
| /external/tensorflow/tensorflow/dtensor/mlir/expansions/ |
| D | elementwise_spmd_expander.cc | 168 const int dim_index = dim_shape_and_index.index(); in ComputeLayoutBackward() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | shape_op.cc | 132 int64_t dim_index; in Compile() local
161 int64_t dim_index; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | attribute_importer.cc | 192 for (int64_t dim_index : xla_layout.minor_to_major()) in ExtractLayoutsFromShapes() 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 | 143 Status RemoteTensorHandleData::Dim(int dim_index, int64_t* dim) const { in Dim()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | dynamic_padder.cc | 377 HloInstruction* dim_index = in GenerateBinaryMask() local
912 for (int64_t dim_index = 0; dim_index < input_window->dimensions_size(); in RewriteInputWithDynamicPadding() local
1186 for (int64_t dim_index = 0; dim_index < rank; ++dim_index) { in RewriteDynamicReduceWindowSamePadding() local
1229 for (int64_t dim_index = 0; dim_index < rank; ++dim_index) { in RewriteDynamicSelectAndScatterSamePadding() local
1274 for (int64_t dim_index = 0; dim_index < rank; ++dim_index) { in RewriteDynamicSelectAndScatterSamePadding() local
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device.cc | 222 int64_t ParallelTensorDim(void* data, int dim_index, TF_Status* status) { in ParallelTensorDim()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 1089 void PrintOneDim(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDim()
1127 void PrintDimSpacing(int dim_index, int num_dims, string* result) { in PrintDimSpacing()
1142 void PrintOneDimV2(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDimV2()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 338 int32_t LiteralBase::GetDynamicSize(int64_t dim_index, in GetDynamicSize()
579 void LiteralBase::Piece::SetDynamicSize(int64_t dim_index, int32_t size) { in SetDynamicSize()
669 void MutableLiteralBase::SetDynamicSize(int64_t dim_index, int32_t size) { in SetDynamicSize()
673 void MutableLiteralBase::SetDynamicSize(int64_t dim_index, in SetDynamicSize()
|
| /external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/ |
| D | Interpreter.cs | 265 private static extern int TfLiteTensorDim(TfLiteTensor tensor, int dim_index); in TfLiteTensorDim()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_tensor.cc | 158 int64_t TF_Dim(const TF_Tensor* t, int dim_index) { in TF_Dim()
|
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper.cc | 150 for (size_t dim_index = 0; dim_index < size; ++dim_index) { in ConvertInputShapeToVector() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_access.cpp | 106 Index first_coeff_index, int dim_index, std::set<Index>* visited_coeffs) { in UpdateCoeffSet()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| D | op_handler_selector_test.cc | 130 Status Dim(int dim_index, int64_t* dim) const override { in Dim()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 49 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index): argument
|