Home
last modified time | relevance | path

Searched defs:float_tensor (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_weights_test.cc171 const Tensor* float_tensor = possible_tensors->Get(i); in FindMatchingExpectedTensor() local
239 const auto float_tensor = FindMatchingExpectedTensor( in TEST_F() local
277 const auto float_tensor = FindMatchingExpectedTensor( in TEST_F() local
647 const auto float_tensor = FindMatchingExpectedTensor( in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_weights_test.cc168 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_F() local
207 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_F() local
581 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_F() local
H A Dmodify_model_interface.cc207 TensorT* float_tensor = subgraph->tensors[tot.input_index].get(); in SetInputTypeToUINT8() local
234 TensorT* float_tensor = subgraph->tensors[tot.output_index].get(); in SetOutputTypeToUINT8() local
H A Dquantize_model_test.cc174 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_P() local
197 const auto float_tensor = float_graph->tensors()->Get(i); in TEST_P() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DDistributionTemplates.h208 auto float_tensor = at::view_as_real(self); in normal_impl_() local
289 auto float_tensor = at::view_as_real(self); in uniform_impl_() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcustom_device_test.cc167 Tensor float_tensor(DT_FLOAT, {}); in TEST() local
H A Dcontext_test.cc269 Tensor float_tensor = test::AsScalar<float>(3.0); in TEST_F() local
/aosp_15_r20/external/federated-compute/fcp/client/engine/
H A Dexample_query_plan_engine_test.cc383 tf::Tensor float_tensor = tensors.value()["float_tensor"]; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dquantize_nodes_test.cc129 const Tensor& float_tensor = float_input.second; in TestGraphWithInputRange() local