Home
last modified time | relevance | path

Searched refs:tensor_elements (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dquantization_util.cc29 float *unpacked_fp32_data, size_t tensor_elements) { in DequantizeFloat16() argument
30 std::transform(packed_fp16_data, packed_fp16_data + tensor_elements, in DequantizeFloat16()
H A Dquantization_util.h50 float* unpacked_fp32_data, size_t tensor_elements);
H A Dxnnpack_delegate.cc4954 size_t tensor_elements = output_tensor.bytes; in PrepareOpsToDelegate() local
4957 tensor_elements /= sizeof(float); in PrepareOpsToDelegate()
4960 tensor_elements /= sizeof(uint16_t); in PrepareOpsToDelegate()
4963 tensor_elements /= sizeof(int8_t); in PrepareOpsToDelegate()
4997 tensor_elements); in PrepareOpsToDelegate()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtf_saved_model_freeze_variables.cc59 TF::ConstOp GetConstOpFromElementsAttr(ElementsAttr tensor_elements, in GetConstOpFromElementsAttr() argument
61 return builder.create<TF::ConstOp>(loc, tensor_elements); in GetConstOpFromElementsAttr()