Searched defs:dim_index (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_debug.cc | 130 int64 dim_index = padded_shape.layout().minor_to_major(i); in TFE_TensorHandleTensorDebugInfo() local
159 TFE_TensorDebugInfo* debug_info, int dim_index) { in TFE_TensorDebugInfoOnDeviceDim()
|
| D | c_api.cc | 466 int64_t TFE_TensorHandleDim(TFE_TensorHandle* h, int dim_index, in TFE_TensorHandleDim()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reduction_ops_common.cc | 106 int dim_index = 0; in Simplify() local
|
| /external/tensorflow/tensorflow/lite/experimental/c/ |
| D | c_api.cc | 172 int32_t TFL_TensorDim(const TFL_Tensor* tensor, int32_t dim_index) { in TFL_TensorDim()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle.cc | 202 Status TensorHandle::Dim(int dim_index, int64* dim) { in Dim()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | tensor.cc | 961 void PrintOneDim(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDim()
999 void PrintDimSpacing(int dim_index, int num_dims, string* result) { in PrintDimSpacing()
1014 void PrintOneDimV2(int dim_index, const gtl::InlinedVector<int64, 4>& shape, in PrintOneDimV2()
|
| D | shape_inference.cc | 716 int64 dim_index = dim_index_in; in ReplaceDim() local
|
| D | common_shape_fns.cc | 1212 int end_value_index, int dim_index) { in ConcatShapeHelper()
|
| /external/skia/modules/skottie/src/ |
| D | SkottieAdapter.cpp | 306 const auto dim_index = SkTPin<size_t>(static_cast<size_t>(fDimensions), in apply() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | array_grad.py | 47 def _ConcatGradHelper(op, grad, start_value_index, end_value_index, dim_index): argument
|
| D | nn_ops.py | 2816 def _swap_axis(logits, dim_index, last_index, name=None): argument
3126 def _move_dim_to_end(tensor, dim_index, rank): argument
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 915 int dim_index; member
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api.cc | 259 int64_t TF_Dim(const TF_Tensor* t, int dim_index) { in TF_Dim()
|