Home
last modified time | relevance | path

Searched defs:input_array (Results 51 – 75 of 79) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Dconvert_reorder_axes.cc103 auto& input_array = model->GetArray(input_array_name); in Run() local
H A Dreorder_reshape_transpose.cc161 const auto& input_array = model->GetArray(input_name); in Run() local
H A Ddequantize.cc196 auto& input_array = model->GetArray(op->inputs[0]); in Run() local
H A Dpropagate_fake_quant_num_bits.cc154 auto& input_array = model->GetArray(input); in RecursivelyBackwardPropagateDataType() local
H A Didentify_nearest_upsample.cc60 bool HasSameValues(const Array& input_array, T value) { in HasSameValues()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/python/
H A Dsaved_model_to_tfl_flatbuffer.cc94 for (const auto& input_array : model_flags.input_arrays()) { in HandleInputOutputArraysWithModule() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dconvolution_dimension_numbers_test.cc88 auto input_array = std::make_unique<Array4D<float>>(2, 3, 5, 5); in XLA_TEST_F() local
H A Dconstants_test.cc138 Array4D<float> input_array(3, 2, 1, 1); in TEST_F() local
H A Dconcat_test.cc761 auto input_array = std::make_unique<Array2D<float>>(4, 2); in XLA_TEST_F() local
H A Dreduce_test.cc822 Array3D<float> input_array(bounds[0], bounds[1], bounds[2]); in XLA_TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/
H A Dmodel_cmdline_flags.cc252 for (const auto& input_array : in ReadModelFlagsFromCommandLineFlags() local
H A Dallocate_transient_arrays.cc241 for (const auto& input_array : model->flags.input_arrays()) { in AllocateTransientArrays() local
H A Dexport_tensorflow.cc194 const auto& input_array = model.GetArray(name); in ConvertFloatTensorConst() local
215 const auto& input_array = model.GetArray(name); in ConvertFloatTensorConst() local
2490 for (const auto& input_array : model.flags.input_arrays()) { in ExportTensorFlowGraphDefImplementation() local
H A Dargs.h128 Arg<std::string> input_array; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dspacetobatch_op_test.py27 def space_to_batch_direct(input_array, block_shape, paddings): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dquantize_op.cc235 TTypes<const float>::Vec input_array = input.flat<float>(); in QuantizeTensor() local
H A Dquantized_concat_op.cc49 typename TTypes<T, 1>::UnalignedConstTensor input_array(src, dims); in Copy() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/logging/
H A Dconversion_log_util.cc178 for (const auto& input_array : model.flags.input_arrays()) { in GetInputAndOutputTypes() local
/aosp_15_r20/external/pdfium/fxbarcode/pdf417/
H A DBC_PDF417HighLevelEncoder_unittest.cpp88 std::vector<uint8_t> input_array; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/android/test/jni/object_tracking/
H A Dobject_tracker_jni.cc438 jbyte* const input_array = env->GetByteArrayElements(input, 0); in OBJECT_TRACKER_METHOD() local
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/array_ops/
H A Dspacetobatch_op_test.py32 def space_to_batch_direct(input_array, block_shape, paddings): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tflite/
H A Dexport.cc488 for (const std::string& input_array : model.GetInvalidInputArrays()) { in Export() local
H A Doperator.cc78 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersioningOpSig() local
1446 const Array& input_array = op_signature.model->GetArray(input_name); in GetVersion() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_concat_op.cc75 typename TTypes<T, 1>::UnalignedConstTensor input_array(src, dims); in Copy() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator_test.cc714 Array4D<float> input_array(3, 2, 1, 1, {1, 2, 3, 4, 5, 6}); in TEST_P() local
756 auto input_array = std::make_unique<Array2D<float>>(4, 3); in TEST_P() local
797 auto input_array = std::make_unique<Array2D<float>>(4, 3); in TEST_P() local

1234