Home
last modified time | relevance | path

Searched defs:input_tensor (Results 126 – 150 of 280) sorted by relevance

12345678910>>...12

/aosp_15_r20/external/federated-compute/fcp/client/engine/
H A Dtflite_plan_engine.cc111 tensorflow::Tensor input_tensor; in RunPlan() local
H A Dsimple_plan_engine.cc144 tensorflow::Tensor input_tensor; in RunPlanInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dresize_bilinear_builder.cc32 const auto& input_tensor = context->tensors[input_tensor_id]; in PopulateSubGraph() local
H A Dstrided_slice_builder.cc31 const auto& input_tensor = context->tensors[inputs->data[0]]; in PopulateSubGraph() local
H A Dslice_builder.cc41 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
H A Dreduce_builder.cc35 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 Dimage_tensor_specs.cc159 const TfLiteTensor* input_tensor = TfLiteEngine::GetInput(&interpreter, 0); in BuildInputImageTensorSpecs() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DQLSTMLayerNormalizationFixture.h61 void fill(InputType &&input_tensor, InputType &&weight_tensor, BiasType &&bias_tensor) in fill()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Das_string_op.cc145 const Tensor* input_tensor; in Compute() local
H A Dregex_replace_op.cc36 const Tensor* input_tensor; in InternalCompute() local
H A Dresource_ops_test.cc128 const Tensor& input_tensor = ctx->input(0); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Duniform_dequantize_op.cc39 auto input_tensor = in EvalPerChannelDequantize() local
H A Duniform_quantize_op.cc42 auto input_tensor = in EvalPerChannelQuantize() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/
H A Dvirtual_device.cc56 void VirtualDeviceContext::CopyTensorInSameDevice(const Tensor* input_tensor, in CopyTensorInSameDevice()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dskip_gram.cc52 const TfLiteTensor* input_tensor; in Prepare() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dnode_file_writer.cc152 AttrValue& input_tensor = in RecordNodeExecution() local
H A Drenamed_device.h114 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 Dngrams_test.py92 def __call__(self, input_tensor): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc969 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 Druntime_fallback_executor.cc188 for (const Tensor& input_tensor : arguments) { in Execute() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmkl_remapper_test.cc104 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 Dmkl_reshape_op.cc61 const Tensor& input_tensor = MklGetInput(context, 0); in Compute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Dtile.cc152 ITensorProxyPtr input_tensor; in Convert() local
H A Dslice_ops.cc152 nvinfer1::ITensor* input_tensor = slice_layer->getInput(0); in HandleDynamicStridedSliceInput() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/signature/
H A Dsignature_def_util_test.cc53 tensorflow::TensorInfo input_tensor; in GetTestSignatureDef() local

12345678910>>...12