Home
last modified time | relevance | path

Searched defs:tensor_b (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/resource/
H A Dresource_variable_test.cc98 TfLiteTensor tensor_a, tensor_b; in TEST() local
139 TfLiteTensor tensor_a, tensor_b; in TEST() local
/aosp_15_r20/external/federated-compute/fcp/aggregation/tensorflow/
H A Dcheckpoint_reader_test.cc44 auto tensor_b = in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dsquared_difference.cc45 const auto& tensor_b = context->tensors[tensor_b_index]; in PopulateSubGraph() local
/aosp_15_r20/external/pytorch/benchmarks/transformer/
H A Dsdp.py199 def assert_close_tensors(tensor_a, tensor_b): argument
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_concat_linear.cpp143 bool isNonZeroDimEqual(Tensor& tensor_a, Tensor& tensor_b) { in isNonZeroDimEqual()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseSemiStructuredLinear.cu622 const auto tensor_b = in _sparse_semi_structured_linear() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_nodes.cc4625 ITensorProxyPtr tensor_b = convert_to_itensor(input_b); in ConvertMatMulImpl() local