| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | insert_logging_test.cc | 49 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 …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | entry.h | 49 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
|
| D | immutable_executor_state.cc | 148 const Tensor* const_tensor = item->kernel->const_tensor(); in Initialize() local
149 if (const_tensor) { in Initialize()
153 const_tensors_.emplace_back(*const_tensor); in Initialize()
155 item->const_tensor = const_tensor; in Initialize()
|
| D | eval_const_tensor.cc | 252 const Tensor* const_tensor = outer_context->input_tensor(index); in ExtractConstantSubgraph() local
253 if (const_tensor) { in ExtractConstantSubgraph()
254 const_inputs->emplace_back(tensor_name, *const_tensor); in ExtractConstantSubgraph()
373 const Tensor* const_tensor = outer_context->input_tensor(index); in EvaluateConstantTensor() local
374 if (const_tensor) { in EvaluateConstantTensor()
|
| D | propagator_debug_utils.cc | 37 return input.const_tensor; in GetTensorValueForDump()
|
| D | executor.cc | 659 output.const_tensor = item.const_tensor; in ProcessConstTensor()
780 } else if (item.const_tensor != nullptr && !params.track_allocations) { in Process()
880 entry->const_tensor = kEmptyTensor; in PrepareInputs()
913 inp->tensor = const_cast<Tensor*>(entry->const_tensor); in PrepareInputs()
|
| D | graph_view.h | 91 const Tensor* const_tensor = nullptr; member
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | single_threaded_executor.cc | 116 const Tensor* const_tensor; in Initialize() local
117 if (n->num_outputs() == 1 && (const_tensor = kernel->const_tensor())) { in Initialize()
127 kernel_state.const_tensor = *const_tensor; in Initialize()
374 input.const_tensor = &kernel_state.const_tensor; in Run()
399 node_inputs[j].tensor = const_cast<Tensor*>(input.const_tensor); in Run()
509 Tensor const_tensor; member
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | constant_op.py | 286 const_tensor = g._create_op_internal( # pylint: disable=protected-access
293 "Const", tuple(), attrs, (const_tensor,), op_name=name, graph=g)
295 const_tensor, = callback_outputs
296 return const_tensor
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 495 Tensor const_tensor; in RegisterConstantNode() local
496 TF_CHECK_OK(MakeTensorFromProto(*proto, &const_tensor)); in RegisterConstantNode()
498 const_node_info.set_dtype(const_tensor.dtype()); in RegisterConstantNode()
500 const_node_info.set_data(const_tensor.tensor_data().data(), data_size); in RegisterConstantNode()
740 Tensor const_tensor; in RegisterPadNode() local
741 TF_CHECK_OK(MakeTensorFromProto(*tensor_proto, &const_tensor)); in RegisterPadNode()
742 CHECK_EQ(2, const_tensor.shape().dims()); in RegisterPadNode()
743 CHECK_EQ(PAD_HEIGHT, const_tensor.shape().dim_size(1)); in RegisterPadNode()
744 if (const_tensor.shape().dim_size(0) == PAD_WIDTH) { in RegisterPadNode()
746 } else if (const_tensor.shape().dim_size(0) < PAD_WIDTH) { in RegisterPadNode()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 1076 auto& const_tensor = *subgraph.tensors[input_num]; in ConvertSubgraph() local
1077 auto const_loc = TensorLoc(const_tensor, builder, base_loc); in ConvertSubgraph()
1080 ? BuildExternalConstOp(const_tensor, const_tensor.buffer, in ConvertSubgraph()
1082 : BuildConstOp(const_tensor, buffers[const_tensor.buffer]->data, in ConvertSubgraph()
1083 const_tensor.is_variable, op_builder, const_loc); in ConvertSubgraph()
1137 auto& const_tensor = *subgraph.tensors[index]; in ConvertSubgraph() local
1138 auto const_loc = TensorLoc(const_tensor, builder, base_loc); in ConvertSubgraph()
1141 ? BuildExternalConstOp(const_tensor, const_tensor.buffer, in ConvertSubgraph()
1143 : BuildConstOp(const_tensor, buffers[const_tensor.buffer]->data, in ConvertSubgraph()
1144 const_tensor.is_variable, op_builder, const_loc); in ConvertSubgraph()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | remote_fused_graph_execute_op_test.cc | 188 Tensor const_tensor; in ExecuteGraph() local
190 *proto, &const_tensor)); in ExecuteGraph()
191 const float b_val = *const_tensor.scalar<float>().data(); in ExecuteGraph()
|
| D | host_constant_op.h | 33 const Tensor* const_tensor() const override { return &tensor_; }; in const_tensor() function
|
| D | constant_op.h | 32 const Tensor* const_tensor() const override { return &tensor_; }; in const_tensor() function
|
| D | resource_variable_ops.h | 28 const Tensor* const_tensor() const override { in const_tensor() function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_launch_util.cc | 347 const Tensor& const_tensor = in SetOutputForConstant() local
350 if (stream && const_tensor.TotalBytes() > 0) { in SetOutputForConstant()
359 ctx->allocate_output(output_num, const_tensor.shape(), &output_tensor)); in SetOutputForConstant()
365 &const_tensor, device, output_tensor, in SetOutputForConstant()
380 ctx->set_output(output_num, const_tensor); in SetOutputForConstant()
|
| D | mark_for_compilation_pass_test.cc | 300 Tensor const_tensor(DT_FLOAT, TensorShape({})); in TEST() local
301 const_tensor.scalar<float>()() = 42.0f; in TEST()
304 .WithAttr("value", const_tensor) in TEST()
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | map_test.py | 1036 const_tensor = ops.get_default_graph().create_op(
1043 return const_tensor
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_v2_test.py | 1609 const_tensor = tf.constant(
1618 return tf.matmul(mult, const_tensor)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.h | 144 virtual const Tensor* const_tensor() const { return nullptr; } in const_tensor() function
|