Home
last modified time | relevance | path

Searched defs:tensor_indices (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/dtensor/cc/
Dsave_restore_util.h55 std::vector<std::vector<int>> tensor_indices; member
Dsave_restore_util.cc137 std::vector<std::vector<int>> tensor_indices; in BuildPerDeviceSave() local
/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer.cc46 const TfLiteIntArray* tensor_indices) { in GetTensorNames()
/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.cc55 const int* tensor_indices, in ValidateTensorIndexingSafe()
77 int max_size, const int* tensor_indices) { in ValidateTensorIndexing()
/external/tensorflow/tensorflow/core/kernels/
Dtensor_array_ops.cc621 const Tensor* tensor_indices; in Compute() local
1062 const Tensor* tensor_indices; in Compute() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibrator.cc276 const std::vector<int>& tensor_indices, in GetLoggableTensorIndices()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder.cc2913 const TfLiteIntArray* tensor_indices, in IsAllAllowedTensors()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc2168 const TfLiteIntArray* tensor_indices, in AppendDynamicDimensions()