/aosp_15_r20/external/federated-compute/fcp/client/engine/ |
H A D | tflite_plan_engine.cc | 111 tensorflow::Tensor input_tensor; in RunPlan() local
|
H A D | simple_plan_engine.cc | 144 tensorflow::Tensor input_tensor; in RunPlanInternal() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
H A D | resize_bilinear_builder.cc | 32 const auto& input_tensor = context->tensors[input_tensor_id]; in PopulateSubGraph() local
|
H A D | strided_slice_builder.cc | 31 const auto& input_tensor = context->tensors[inputs->data[0]]; in PopulateSubGraph() local
|
H A D | slice_builder.cc | 41 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
|
H A D | reduce_builder.cc | 35 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
|
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/vision/utils/ |
H A D | image_tensor_specs.cc | 159 const TfLiteTensor* input_tensor = TfLiteEngine::GetInput(&interpreter, 0); in BuildInputImageTensorSpecs() local
|
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
H A D | QLSTMLayerNormalizationFixture.h | 61 void fill(InputType &&input_tensor, InputType &&weight_tensor, BiasType &&bias_tensor) in fill()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
H A D | as_string_op.cc | 145 const Tensor* input_tensor; in Compute() local
|
H A D | regex_replace_op.cc | 36 const Tensor* input_tensor; in InternalCompute() local
|
H A D | resource_ops_test.cc | 128 const Tensor& input_tensor = ctx->input(0); in Compute() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/ |
H A D | uniform_dequantize_op.cc | 39 auto input_tensor = in EvalPerChannelDequantize() local
|
H A D | uniform_quantize_op.cc | 42 auto input_tensor = in EvalPerChannelQuantize() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
H A D | virtual_device.cc | 56 void VirtualDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | skip_gram.cc | 52 const TfLiteTensor* input_tensor; in Prepare() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
H A D | node_file_writer.cc | 152 AttrValue& input_tensor = in RecordNodeExecution() local
|
H A D | renamed_device.h | 114 void CopyTensorInSameDevice(const Tensor* input_tensor, Tensor* output_tensor, in CopyTensorInSameDevice()
|
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ |
H A D | ngrams_test.py | 92 def __call__(self, input_tensor): argument
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/ |
H A D | subgraph.cc | 969 TfLiteTensor* input_tensor = tensor(input_tensor_idx); in ReleaseMemory() local 1973 TfLiteTensor* input_tensor = tensor(input_tensor_index); in InitializeTensorReleaseMap() local 2009 TfLiteTensor* input_tensor = tensor(input_tensor_index); in MaybeReleaseDynamicTensors() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/runtime_fallback/ |
H A D | runtime_fallback_executor.cc | 188 for (const Tensor& input_tensor : arguments) { in Execute() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
H A D | mkl_remapper_test.cc | 104 auto input_tensor = GenerateRandomTensor<DT_FLOAT>( in FuseConv2DWithBiasAndAddNOrAdd() local 480 auto input_tensor = GenerateRandomTensor<DT_FLOAT>( in TEST_F() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
H A D | mkl_reshape_op.cc | 61 const Tensor& input_tensor = MklGetInput(context, 0); in Compute() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/ |
H A D | tile.cc | 152 ITensorProxyPtr input_tensor; in Convert() local
|
H A D | slice_ops.cc | 152 nvinfer1::ITensor* input_tensor = slice_layer->getInput(0); in HandleDynamicStridedSliceInput() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/signature/ |
H A D | signature_def_util_test.cc | 53 tensorflow::TensorInfo input_tensor; in GetTestSignatureDef() local
|