Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Dvariable_ops.cc49 auto tensor_flat = tensor.flat<T>(); in ReadVariableHelper() local
181 auto tensor_flat = tensor.flat<T>(); in ValidateImpl() local
279 auto tensor_flat = tensor.flat<T>(); in ValidateImpl() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/
H A Dnn_grad_test.cc91 auto tensor_flat = tensor->flat<T>(); in SetRandomValuesForMaxPooling() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes_test.cc2286 auto tensor_flat = tensor_ptr->flat<CType>(); in TestConvertVariableV2() local
2378 auto tensor_flat = resource->tensor()->flat<CType>(); in TestConvertReadVariableOp() local