| /external/tensorflow/tensorflow/python/debug/lib/ |
| D | debug_v2_ops_test.py | 342 tensor_id = np.power(2, 53, dtype=np.int64)
348 tensor_id=tensor_id,
351 tensor_id += 1
358 tensor_id=tensor_id,
369 tensor_id=x._id,
372 tensor, tensor_id = debug_summary(constant_op.constant([]))
373 self.assertAllEqual(tensor, [tensor_id, 0.0])
375 tensor, tensor_id = debug_summary(constant_op.constant(42.0))
376 self.assertAllEqual(tensor, [tensor_id, 0.0])
378 tensor, tensor_id = debug_summary(constant_op.constant([3.0, 4.0]))
[all …]
|
| D | dumping_callback_test.py | 300 tensor_id = reader.graph_execution_trace_to_tensor_id(trace)
301 self.assertGreaterEqual(tensor_id, 0)
303 self.assertAllClose(trace.debug_tensor_value, [tensor_id, 1])
305 self.assertAllClose(trace.debug_tensor_value, [tensor_id, 0])
311 tensor_id = reader.graph_execution_trace_to_tensor_id(trace)
312 self.assertGreaterEqual(tensor_id, 0)
315 [tensor_id, 8, 1, 3, 2])
318 [tensor_id, 8, 0, 0, 0])
327 tensor_id = reader.graph_execution_trace_to_tensor_id(trace)
328 self.assertGreaterEqual(tensor_id, 0)
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | op_builder.cc | 136 const TensorID& tensor_id) { in AddTensorWithID() argument
141 tensors_[tf_tensor_id] = tensor_id; in AddTensorWithID()
144 std::string GraphBuilder::GetTensorName(int tensor_id) { in GetTensorName() argument
145 return GetTensorID(tensor_id).ToString(); in GetTensorName()
148 const TensorID GraphBuilder::GetTensorID(int tensor_id) { in GetTensorID() argument
149 if (!HasTensor(tensor_id)) { in GetTensorID()
153 tensor_id, static_cast<int>(tensors_.size())); in GetTensorID()
157 used_tensor_[tensor_id] = true; in GetTensorID()
158 return tensors_[tensor_id]; in GetTensorID()
218 auto tensor_id = TensorID(GetID(), num_outputs_++); in AddOutput() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | pad_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
33 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
34 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
40 tensor_id = inputs->data[1]; in PopulateSubGraph()
41 const auto& padding_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
46 graph_builder_->AddConstNodeWithData(tensor_id, padding_tensor); in PopulateSubGraph()
49 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| D | resize_nearest_neighbor_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
33 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
34 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
37 tensor_id = inputs->data[1]; in PopulateSubGraph()
38 const auto& output_dim_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
43 graph_builder_->AddConstNodeWithData(tensor_id, output_dim_tensor); in PopulateSubGraph()
46 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| D | op_builder.cc | 201 OpBuilder* GraphBuilder::AddConstNodeWithData(int tensor_id, in AddConstNodeWithData() argument
215 if (!HasTensor(tensor_id)) in AddConstNodeWithData()
216 AddTensorWithID(tensor_id, lookup_result->GetID(), 0); in AddConstNodeWithData()
228 context_, "Failed to add const node for tensor with id: %d", tensor_id); in AddConstNodeWithData()
231 AddTensorWithID(tensor_id, node_id, 0); in AddConstNodeWithData()
240 AddCastOp(context_, OP_Quantized_CastInt8ToUInt8, tensor_id, in AddConstNodeWithData()
250 int tensor_id, in AddCastOp() argument
259 tensor_data->data[0] = tensor_id; in AddCastOp()
279 const int tensor_id = input_tensors->data[i]; in AddInputTensors() local
280 const auto& tensor = context->tensors[tensor_id]; in AddInputTensors()
[all …]
|
| D | reduce_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
34 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
35 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
39 tensor_id = inputs->data[1]; in PopulateSubGraph()
40 const auto& axes_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
45 graph_builder_->AddConstNodeWithData(tensor_id, axes_tensor); in PopulateSubGraph()
|
| D | arithmetic_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
34 const auto& input1_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
35 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
38 tensor_id = inputs->data[1]; in PopulateSubGraph()
39 const auto& input2_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
40 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| D | mirror_pad_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
34 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
35 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
39 tensor_id = inputs->data[1]; in PopulateSubGraph()
40 const auto& padding_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
|
| D | transpose_conv_2d_builder.cc | 43 int tensor_id = inputs->data[2]; in PopulateSubGraph() local
44 const auto& data_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
45 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
48 tensor_id = inputs->data[1]; in PopulateSubGraph()
49 const auto& weights_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
67 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| D | neg_op_builder.cc | 26 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
27 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
28 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| D | l2_normalization_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
33 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
34 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| D | transpose_builder.cc | 28 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
29 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
30 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| D | hardswish_builder.cc | 31 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
32 const auto& input1_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
33 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| D | space_to_depth_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
42 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
45 ComputeAndAddMinAndMax(context, context->tensors[tensor_id])); in PopulateSubGraph()
|
| D | activation_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
34 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
35 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| D | softmax_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
34 const auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
35 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| D | pack_builder.cc | 54 int tensor_id = -1; in PopulateSubGraph() local
57 tensor_id = inputs->data[i]; in PopulateSubGraph()
58 auto& input_tensor = context->tensors[tensor_id]; in PopulateSubGraph()
59 AddInput(graph_builder_->GetHexagonTensorId(tensor_id)); in PopulateSubGraph()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | tensor_id_test.cc | 95 TensorId tensor_id = ParseTensorName(input); in TEST() local
96 EXPECT_TRUE(IsTensorIdControl(tensor_id)); in TEST()
99 tensor_id = ParseTensorName(input); in TEST()
100 EXPECT_FALSE(IsTensorIdControl(tensor_id)); in TEST()
103 tensor_id = ParseTensorName(input); in TEST()
104 EXPECT_FALSE(IsTensorIdControl(tensor_id)); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view.h | 159 TensorId tensor_id = ParseTensorName(port.node->input(i)); in GetFanin() local
161 auto it = nodes_.find(tensor_id.node()); in GetFanin()
162 if (it != nodes_.end()) result.emplace(it->second, tensor_id.index()); in GetFanin()
177 TensorId tensor_id = ParseTensorName(port.node->input(port.port_id)); in GetRegularFanin() local
178 return GetOutputPort(tensor_id.node(), tensor_id.index()); in GetRegularFanin()
236 TensorId tensor_id = ParseTensorName(node.input(i)); in GetFanins() local
238 auto it = nodes_.find(tensor_id.node()); in GetFanins()
239 if (it != nodes_.end()) result.emplace(it->second, tensor_id.index()); in GetFanins()
307 TensorId tensor_id = ParseTensorName(node.input(i)); in GetFaninEdges() local
309 auto it = nodes_.find(tensor_id.node()); in GetFaninEdges()
[all …]
|
| D | mutable_graph_view.cc | 43 bool IsTensorIdPortValid(const TensorId& tensor_id) { in IsTensorIdPortValid() argument
44 return tensor_id.index() >= Graph::kControlSlot; in IsTensorIdPortValid()
47 bool IsTensorIdRegular(const TensorId& tensor_id) { in IsTensorIdRegular() argument
48 return tensor_id.index() > Graph::kControlSlot; in IsTensorIdRegular()
51 bool IsTensorIdControlling(const TensorId& tensor_id) { in IsTensorIdControlling() argument
52 return tensor_id.index() == Graph::kControlSlot; in IsTensorIdControlling()
65 TensorId tensor_id = ParseTensorName(node.input(0)); in IsIdentityConsumingSwitch() local
66 if (IsTensorIdControlling(tensor_id)) { in IsIdentityConsumingSwitch()
70 NodeDef* input_node = graph.GetNode(tensor_id.node()); in IsIdentityConsumingSwitch()
123 TensorId tensor_id = ParseTensorName(control_fanout.node->input(i)); in SwapControlledFanoutInputs() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 372 for (auto& tensor_id : preallocated_ids) { in AllocateTensors() local
373 const auto& t = tensor_reserver_.Get(tensor_id); in AllocateTensors()
375 nil, t.shape, t.descriptor, &preallocated_tensors_[tensor_id])); in AllocateTensors()
384 MetalSpatialTensor* InferenceContext::GetTensor(ValueId tensor_id) { in GetTensor() argument
385 if (preallocated_tensors_.find(tensor_id) != preallocated_tensors_.end()) { in GetTensor()
386 return &preallocated_tensors_[tensor_id]; in GetTensor()
387 } else if (const_tensors_.find(tensor_id) != const_tensors_.end()) { in GetTensor()
388 return &const_tensors_[tensor_id]; in GetTensor()
389 } else if (graph_ids_to_shared_buffer_tensors_.find(tensor_id) != in GetTensor()
392 [graph_ids_to_shared_buffer_tensors_[tensor_id]]; in GetTensor()
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | graph_optimizer_stage.cc | 52 SafeTensorId tensor_id = ParseTensorName(tensor); in GetTensorProperties() local
54 if (tensor_id.index() < 0) { in GetTensorProperties()
60 ctx.graph_properties->GetOutputProperties(tensor_id.node()); in GetTensorProperties()
63 if (num_outputs == 0 || tensor_id.index() > num_outputs - 1) { in GetTensorProperties()
65 "Node ", tensor_id.node(), in GetTensorProperties()
66 " is missing output properties at position :", tensor_id.index(), in GetTensorProperties()
70 *properties = &output_properties[tensor_id.index()]; in GetTensorProperties()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 149 void Watch(int64 tensor_id);
164 void DeleteTrace(int64 tensor_id);
249 void Watch(int64 tensor_id, Gradient* tangent);
253 void DeleteGradient(int64 tensor_id);
303 Gradient* FetchJVP(int64 tensor_id);
401 int64 tensor_id) { in Watch() argument
402 tensor_tape_.emplace(tensor_id, -1); in Watch()
438 int64 tensor_id) { in DeleteTrace() argument
439 auto it = tensor_usage_.find(tensor_id); in DeleteTrace()
448 auto tensor_op_it = tensor_tape_.find(tensor_id); in DeleteTrace()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 88 const int tensor_id = input_tensors->data[i];
89 const auto& tensor = context->tensors[tensor_id];
90 builder_->AddTensorWithID(tensor_id, delegates::coreml::TensorID(0, num_inputs++));
98 const int tensor_id = output_tensors->data[i];
99 const auto& tensor = context->tensors[tensor_id];
102 output->set_name(builder_->GetTensorName(tensor_id));
152 const int tensor_id = delegate_params->input_tensors->data[i];
153 if (builder_->IsTensorUsed(tensor_id)) {
154 const auto& tensor = context->tensors[tensor_id];
156 input->set_name(builder_->GetTensorName(tensor_id));
[all …]
|