Home
last modified time | relevance | path

Searched defs:tensor_length (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dcollective_nccl_test.cc463 void InitExpected(std::vector<float>* expected, const int tensor_length, in InitExpected()
498 void InitExpected(std::vector<float>* expected, const int tensor_length, in InitExpected()
532 void InitExpected(std::vector<float>* expected, const int tensor_length, in InitExpected()
H A Dragged_tensor_to_tensor_op.cc106 const INDEX_TYPE tensor_length = row_split.size(); in GetMaxWidthRowSplit() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
H A Dragged_tensor_to_tensor_tflite.cc188 const int tensor_length = tensor_shape.FlatSize(); in GetMaxWidthRowSplit() local