| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | op_builder.cc | 136 const TensorID& tensor_id) { in AddTensorWithID()
144 std::string GraphBuilder::GetTensorName(int tensor_id) { in GetTensorName()
148 const TensorID GraphBuilder::GetTensorID(int tensor_id) { in GetTensorID()
218 auto tensor_id = TensorID(GetID(), num_outputs_++); in AddOutput() local
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | tape.h | 401 int64 tensor_id) { in Watch()
438 int64 tensor_id) { in DeleteTrace()
534 int64 tensor_id = tensor_stack.back(); in PrepareBackprop() local
873 int64 tensor_id = source_tensor_ids[i]; in ComputeGradient() local
1001 int64 tensor_id = vspace_.TensorId(grad_tensor); in ForwardpropFromTape() local
1098 int64 tensor_id = output_tensors[i].GetID(); in Accumulate() local
1117 int64 tensor_id, Gradient* tangent) { in Watch()
1135 int64 tensor_id) { in DeleteGradient()
1145 int64 tensor_id) { in FetchJVP()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | graph_view.h | 159 TensorId tensor_id = ParseTensorName(port.node->input(i)); in GetFanin() local
177 TensorId tensor_id = ParseTensorName(port.node->input(port.port_id)); in GetRegularFanin() local
236 TensorId tensor_id = ParseTensorName(node.input(i)); in GetFanins() local
307 TensorId tensor_id = ParseTensorName(node.input(i)); in GetFaninEdges() local
340 TensorId tensor_id = ParseTensorName(node->input(i)); in AddFanouts() local
|
| D | mutable_graph_view.cc | 43 bool IsTensorIdPortValid(const TensorId& tensor_id) { in IsTensorIdPortValid()
47 bool IsTensorIdRegular(const TensorId& tensor_id) { in IsTensorIdRegular()
51 bool IsTensorIdControlling(const TensorId& tensor_id) { in IsTensorIdControlling()
65 TensorId tensor_id = ParseTensorName(node.input(0)); in IsIdentityConsumingSwitch() local
123 TensorId tensor_id = ParseTensorName(control_fanout.node->input(i)); in SwapControlledFanoutInputs() local
349 TensorId tensor_id = ParseTensorName(node->input(pos)); in AddAndDedupFanouts() local
655 TensorId tensor_id = ParseTensorName(node->input(i)); in SwapNodeNames() local
936 TensorId tensor_id = ParseTensorName(node->input(i)); in AddRegularFaninByPort() local
964 TensorId tensor_id(fanin.node->name(), fanin.port_id); in GetControllingFaninToAdd() local
1067 TensorId tensor_id = ParseTensorName(node->input(i)); in RemoveRegularFaninInternal() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | op_builder.cc | 201 OpBuilder* GraphBuilder::AddConstNodeWithData(int tensor_id, in AddConstNodeWithData()
250 int tensor_id, in AddCastOp()
279 const int tensor_id = input_tensors->data[i]; in AddInputTensors() local
301 const int tensor_id = output_tensors->data[i]; in AddOutputTensors() local
|
| D | transpose_builder.cc | 28 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | neg_op_builder.cc | 26 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | hardswish_builder.cc | 31 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | l2_normalization_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | space_to_depth_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | activation_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | pad_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | softmax_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | cast_builder.cc | 41 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | mirror_pad_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | resize_nearest_neighbor_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | pool_2d_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | arithmetic_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | reduce_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| D | slice_builder.cc | 40 const int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 372 for (auto& tensor_id : preallocated_ids) { in AllocateTensors() local
384 MetalSpatialTensor* InferenceContext::GetTensor(ValueId tensor_id) { in GetTensor()
450 for (auto& tensor_id : nodes_[op_index].inputs) { in GetUsages() local
455 for (auto& tensor_id : nodes_[op_index].outputs) { in GetUsages() local
528 for (auto& tensor_id : all_ids) { in AllocateMemoryForBuffers() local
569 for (auto& tensor_id : all_ids) { in AllocateMemoryForStrongShapes() local
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | utils.cc | 32 bool TensorTypeMatch(int tensor_id, TfLiteContext* context, in TensorTypeMatch()
124 int tensor_id; in IsNodeSupportedByHexagon() local
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | tensor_id.cc | 65 bool IsTensorIdControl(const TensorId& tensor_id) { in IsTensorIdControl()
|
| D | tensor_id_test.cc | 95 TensorId tensor_id = ParseTensorName(input); in TEST() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | deadness_analysis.cc | 283 explicit SymbolPredicate(int64 id, TensorId tensor_id, bool must_be_true) in SymbolPredicate()
301 TensorId tensor_id() const { return tensor_id_; } in tensor_id() function in tensorflow::__anonaeebdbfa0111::SymbolPredicate
316 explicit IntSymbolPredicate(int64 id, TensorId tensor_id, in IntSymbolPredicate()
338 TensorId tensor_id() const { return tensor_id_; } in tensor_id() function in tensorflow::__anonaeebdbfa0111::IntSymbolPredicate
420 TensorId tensor_id(node->name(), output_idx); in MakeSymbolPredicate() local
456 TensorId tensor_id(node->name(), output_idx); in MakeSymbolPredicate() local
1559 for (TensorId tensor_id : tensor_ids) { in Print() local
|