Home
last modified time | relevance | path

Searched defs:const_tensor (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dinsert_logging_test.cc49 Tensor const_tensor(DT_FLOAT, TensorShape({10})); in TestInsertLogging() local
96 Tensor const_tensor(DT_FLOAT, TensorShape({10})); in TestInsertLoggingByOpType() local
145 Tensor const_tensor(DT_FLOAT, TensorShape({10})); in TestInsertLoggingByPrefix() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dcombine_tpu_embedding_load_retrieve_pass.cc520 TensorProto const_tensor = in Run()
538 TensorProto const_tensor = in Run() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Deval_const_tensor.cc404 const Tensor* const_tensor = outer_context->input_tensor(index); in ExtractConstantSubgraph() local
536 const Tensor* const_tensor = outer_context->input_tensor(index); in EvaluateConstantTensor() local
H A Dsingle_threaded_executor.cc130 const Tensor* const_tensor; in Initialize() local
553 Tensor const_tensor; member
H A Dentry.h119 const Tensor* const_tensor; member
H A Dimmutable_executor_state.cc149 const Tensor* const_tensor = item->kernel->const_tensor(); in Initialize() local
H A Dgraph_view.h93 const Tensor* const_tensor = nullptr; member
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconstant_op.h32 const Tensor* const_tensor() const override { return &tensor_; }; in const_tensor() function
H A Dhost_constant_op.h33 const Tensor* const_tensor() const override { return &tensor_; }; in const_tensor() function
H A Dresource_variable_ops.h28 const Tensor* const_tensor() const override { in const_tensor() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A Dkernel.cpp61 auto const_tensor = toIValue(value)->toTensor(); in initializeConstantInputs() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_import.cc1313 auto& const_tensor = *subgraph.tensors[input_num]; in ConvertSubgraph() local
1370 auto& const_tensor = *subgraph.tensors[index]; in ConvertSubgraph() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dgraph_opt.cpp316 auto const_tensor = toIValue(v)->toTensor(); in fixupTypeInfoForValue() local
H A Dkernel.cpp1438 auto const_tensor = toIValue(v)->toTensor(); in bindConstant() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dgraph_partition.cc179 Tensor const_tensor(data_type, const_shape); in PrepareSubgraphForFunctionConversion() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_launch_util.cc382 const Tensor& const_tensor = in SetOutputForConstant() local
H A Dmark_for_compilation_pass_test.cc339 Tensor const_tensor(DT_FLOAT, TensorShape({})); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc1088 Tensor const_tensor; in VerifyConstants() local
1547 Tensor const_tensor; in FindMklLayerNorm() local
2528 Tensor const_tensor; in AddFusedConv3DNode() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_ops_to_mkldnn.cpp950 auto const_tensor = constant_as<Tensor>(v); in tensorInputIsMKLDNNSupported() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel.h155 virtual const Tensor* const_tensor() const { return nullptr; } in const_tensor() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder.cc811 const TfLiteTensor* const_tensor = reader->GetInputTensor(0); in Parse() local