Home
last modified time | relevance | path

Searched refs:const_tensor (Results 1 – 25 of 33) sorted by relevance

12

/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
50 test::FillIota<float>(&const_tensor, 1.0f); in TestInsertLogging()
52 Const(root.WithOpName("const_node1"), Input::Initializer(const_tensor)); in TestInsertLogging()
54 Const(root.WithOpName("const_node2"), Input::Initializer(const_tensor)); in TestInsertLogging()
56 Const(root.WithOpName("const_node3"), Input::Initializer(const_tensor)); in TestInsertLogging()
96 Tensor const_tensor(DT_FLOAT, TensorShape({10})); in TestInsertLoggingByOpType() local
97 test::FillIota<float>(&const_tensor, 1.0f); in TestInsertLoggingByOpType()
99 Const(root.WithOpName("const_node1"), Input::Initializer(const_tensor)); in TestInsertLoggingByOpType()
101 Const(root.WithOpName("const_node2"), Input::Initializer(const_tensor)); in TestInsertLoggingByOpType()
103 Const(root.WithOpName("const_node3"), Input::Initializer(const_tensor)); in TestInsertLoggingByOpType()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dentry.h49 const_tensor = other.const_tensor; in Entry()
74 const_tensor = other.const_tensor;
96 const_tensor = other.const_tensor;
119 const Tensor* const_tensor; member
H A Dsingle_threaded_executor.cc130 const Tensor* const_tensor; in Initialize() local
131 if (n->num_outputs() == 1 && (const_tensor = kernel->const_tensor())) { in Initialize()
141 kernel_state.const_tensor = *const_tensor; in Initialize()
403 input.const_tensor = &kernel_state.const_tensor; in Run()
428 node_inputs[j].tensor = const_cast<Tensor*>(input.const_tensor); in Run()
553 Tensor const_tensor; member
H A Dimmutable_executor_state.cc149 const Tensor* const_tensor = item->kernel->const_tensor(); in Initialize() local
150 if (const_tensor) { in Initialize()
154 const_tensors_.emplace_back(*const_tensor); in Initialize()
156 item->const_tensor = const_tensor; in Initialize()
H A Deval_const_tensor.cc404 const Tensor* const_tensor = outer_context->input_tensor(index); in ExtractConstantSubgraph() local
405 if (const_tensor) { in ExtractConstantSubgraph()
406 const_inputs->emplace_back(tensor_name, *const_tensor); in ExtractConstantSubgraph()
536 const Tensor* const_tensor = outer_context->input_tensor(index); in EvaluateConstantTensor() local
537 if (const_tensor) { in EvaluateConstantTensor()
H A Dexecutor.cc687 output.const_tensor = item.const_tensor; in ProcessConstTensor()
832 } else if (item.const_tensor != nullptr && !params.track_allocations) { in ProcessInline()
937 entry->const_tensor = kEmptyTensor; in PrepareInputs()
970 inp->tensor = const_cast<Tensor*>(entry->const_tensor); in PrepareInputs()
H A Dpropagator_debug_utils.cc37 return input.const_tensor; in GetTensorValueForDump()
H A Dgraph_view.h93 const Tensor* const_tensor = nullptr; member
/aosp_15_r20/external/pytorch/test/export/
H A Dtest_lift_unlift.py185 const_tensor = builder.values["const_tensor"]
188 root = {"const_tensor": const_tensor, "const_obj": const_obj}
202 self.assertEqual(constants["lifted_tensor_0"], const_tensor)
218 self.assertEqual(lifted_tensor_placeholder.meta["val"], const_tensor)
259 const_tensor = builder.values["const_tensor"]
272 self.assertEqual(constants["foo.lifted_tensor_0"], const_tensor)
397 const_tensor = torch.ones(2, 3)
399 constant_attr_map.add(const_tensor, "foo.bar.baz")
401 self.assertEqual(list(constant_attr_map), [const_obj, const_tensor])
402 self.assertEqual(list(constant_attr_map.keys()), [const_obj, const_tensor])
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_import.cc1313 auto& const_tensor = *subgraph.tensors[input_num]; in ConvertSubgraph() local
1314 auto const_loc = TensorLoc(const_tensor, builder, base_loc); in ConvertSubgraph()
1317 ? BuildExternalConstOp(const_tensor, const_tensor.buffer, in ConvertSubgraph()
1319 : BuildConstOp(const_tensor, buffers[const_tensor.buffer]->data, in ConvertSubgraph()
1320 const_tensor.is_variable, op_builder, const_loc); in ConvertSubgraph()
1370 auto& const_tensor = *subgraph.tensors[index]; in ConvertSubgraph() local
1371 auto const_loc = TensorLoc(const_tensor, builder, base_loc); in ConvertSubgraph()
1374 ? BuildExternalConstOp(const_tensor, const_tensor.buffer, in ConvertSubgraph()
1376 : BuildConstOp(const_tensor, buffers[const_tensor.buffer]->data, in ConvertSubgraph()
1377 const_tensor.is_variable, op_builder, const_loc); in ConvertSubgraph()
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconstant_op.py289 const_tensor = g._create_op_internal( # pylint: disable=protected-access
296 "Const", tuple(), attrs, (const_tensor,), op_name=name, graph=g)
298 const_tensor, = callback_outputs
299 return const_tensor
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dcombine_tpu_embedding_load_retrieve_pass.cc500 TensorProto const_tensor, in Run()
504 .Attr("value", const_tensor) in Run()
520 TensorProto const_tensor = in Run() local
524 make_const_node(const_node_name, const_tensor, &constant_node)); in Run()
538 TensorProto const_tensor = in Run() local
542 make_const_node(const_node_name, const_tensor, &constant_node)); in Run()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dxla_launch_util.cc382 const Tensor& const_tensor = in SetOutputForConstant() local
385 if (stream && const_tensor.TotalBytes() > 0) { in SetOutputForConstant()
394 ctx->allocate_output(output_num, const_tensor.shape(), &output_tensor)); in SetOutputForConstant()
400 &const_tensor, device, output_tensor, in SetOutputForConstant()
415 ctx->set_output(output_num, const_tensor); in SetOutputForConstant()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc1088 Tensor const_tensor; in VerifyConstants() local
1090 const_tensor.FromProto(node_def->attr().at("value").tensor())) { in VerifyConstants()
1091 if (const_tensor.NumElements() == 1) { in VerifyConstants()
1092 DataType dtype = const_tensor.dtype(); in VerifyConstants()
1095 const_value = const_tensor.flat<float>()(0); in VerifyConstants()
1097 const_value = const_tensor.flat<bfloat16>()(0); in VerifyConstants()
1099 const_value = const_tensor.flat<Eigen::half>()(0); in VerifyConstants()
1547 Tensor const_tensor; in FindMklLayerNorm() local
1549 const_tensor.FromProto(empty_const_node->attr().at("value").tensor())) { in FindMklLayerNorm()
1550 if (const_tensor.NumElements() != 0) return false; in FindMklLayerNorm()
[all …]
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Dkernel.cpp1438 auto const_tensor = toIValue(v)->toTensor(); in bindConstant() local
1439 auto scalar_type = c10::typeMetaToScalarType(const_tensor.options().dtype()); in bindConstant()
1440 auto sizes = const_tensor.sizes(); in bindConstant()
1451 if (!const_tensor.is_contiguous()) { in bindConstant()
1452 const_tensor = const_tensor.clone().contiguous(); in bindConstant()
1453 unpacked_constant_tensors_.push_back(const_tensor); in bindConstant()
1456 constants_.push_back({buf, const_tensor.data_ptr()}); in bindConstant()
H A Dgraph_opt.cpp316 auto const_tensor = toIValue(v)->toTensor(); in fixupTypeInfoForValue() local
318 tensorTypeInCurrentExecutionContext(const_tensor); in fixupTypeInfoForValue()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dgraph_partition.cc179 Tensor const_tensor(data_type, const_shape); in PrepareSubgraphForFunctionConversion() local
180 const_tensor.flat<int>()(0) = 0; in PrepareSubgraphForFunctionConversion()
186 .Attr("value", const_tensor) in PrepareSubgraphForFunctionConversion()
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/onednn/
H A Dkernel.cpp61 auto const_tensor = toIValue(value)->toTensor(); in initializeConstantInputs() local
62 constantInputs_.emplace_back(const_tensor); in initializeConstantInputs()
/aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/_tensor/
H A D__init__.py4 from .const_tensor import ConstTensor
H A Dworkload_tensors.py13 from .const_tensor import ConstTensor
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dhost_constant_op.h33 const Tensor* const_tensor() const override { return &tensor_; }; in const_tensor() function
H A Dconstant_op.h32 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/passes/
H A Dfrozen_ops_to_mkldnn.cpp950 auto const_tensor = constant_as<Tensor>(v); in tensorInputIsMKLDNNSupported() local
951 if (const_tensor) { in tensorInputIsMKLDNNSupported()
952 return supportedMKLDNNWeight(*const_tensor); in tensorInputIsMKLDNNSupported()
/aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/
H A D__init__.py116 from ._tensor.const_tensor import ConstTensor

12