| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | mutable_graph_view.cc | 44 bool IsTensorIdPortValid(const TensorId& tensor_id) { in IsTensorIdPortValid()
48 bool IsTensorIdRegular(const TensorId& tensor_id) { in IsTensorIdRegular()
52 bool IsTensorIdControlling(const TensorId& tensor_id) { in IsTensorIdControlling()
66 TensorId tensor_id = ParseTensorName(node.input(0)); in IsIdentityConsumingSwitch() local
127 TensorId tensor_id = ParseTensorName(control_fanout.node->input(i)); in SwapControlledFanoutInputs() local
371 TensorId tensor_id = ParseTensorName(node->input(pos)); in AddAndDedupFanouts() local
677 TensorId tensor_id = ParseTensorName(node->input(i)); in SwapNodeNames() local
958 TensorId tensor_id = ParseTensorName(node->input(i)); in AddRegularFaninByPort() local
986 TensorId tensor_id(fanin.node->name(), fanin.port_id); in GetControllingFaninToAdd() local
1089 TensorId tensor_id = ParseTensorName(node->input(i)); in RemoveRegularFaninInternal() local
[all …]
|
| H A 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
|
| H A D | utils.cc | 76 string TensorIdToString(const TensorId& tensor_id) { in TensorIdToString()
81 string SafeTensorIdToString(const SafeTensorId& tensor_id) { in SafeTensorIdToString()
96 bool IsControlInput(const TensorId& tensor_id) { return tensor_id.index() < 0; } in IsControlInput()
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/eager/ |
| H A D | tape.h | 401 int64_t tensor_id) { in Watch()
438 int64_t tensor_id) { in DeleteTrace()
534 int64_t tensor_id = tensor_stack.back(); in PrepareBackprop() local
873 int64_t tensor_id = source_tensor_ids[i]; in ComputeGradient() local
1001 int64_t tensor_id = vspace_.TensorId(grad_tensor); in ForwardpropFromTape() local
1098 int64_t tensor_id = output_tensors[i].GetID(); in Accumulate() local
1117 int64_t tensor_id, Gradient* tangent) { in Watch()
1135 int64_t tensor_id) { in DeleteGradient()
1145 int64_t tensor_id) { in FetchJVP()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| H A D | op_builder.cc | 138 const TensorID& tensor_id) { in AddTensorWithID()
146 std::string GraphBuilder::GetTensorName(int tensor_id) { in GetTensorName()
150 const TensorID GraphBuilder::GetTensorID(int tensor_id) { in GetTensorID()
220 auto tensor_id = TensorID(GetID(), num_outputs_++); in AddOutput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A 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
|
| H A D | neg_op_builder.cc | 26 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| H A D | transpose_builder.cc | 28 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| H A D | l2_normalization_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| H A D | hardswish_builder.cc | 31 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| H A D | space_to_depth_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| H A D | activation_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| H A D | pad_builder.cc | 32 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| H A D | softmax_builder.cc | 33 int tensor_id = inputs->data[0]; in PopulateSubGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| H A D | operation_selector.cc | 208 int tensor_id = gpu_subgraph->AddTensor( in AddDynamicConv() local
217 int tensor_id = gpu_subgraph->AddTensor( in AddDynamicConv() local
285 int tensor_id = gpu_subgraph->AddTensor(std::move(weights_tensor)); in AddConvSharedWeights() local
301 int tensor_id = gpu_subgraph->AddTensor(std::move(weights_tensor)); in AddConvSharedWeights() local
430 int tensor_id = gpu_subgraph->AddTensor(concatenated_shape, in GPUOperationFromNodePart0() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| H A D | inference_context.cc | 324 MetalSpatialTensor* InferenceContext::GetTensor(ValueId tensor_id) { in GetTensor()
423 for (auto& tensor_id : nodes_[op_index].inputs) { in GetUsages() local
428 for (auto& tensor_id : nodes_[op_index].outputs) { in GetUsages() local
561 for (auto& tensor_id : all_ids) { in AllocateMemoryForBuffers() local
632 for (auto& tensor_id : all_ids) { in AllocateMemoryForStrongShapes() local
771 absl::Status InferenceContext::SetTensor(const ValueId& tensor_id, in SetTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | deadness_analysis.cc | 283 explicit SymbolPredicate(int64_t id, TensorId tensor_id, bool must_be_true) in SymbolPredicate()
301 TensorId tensor_id() const { return tensor_id_; } in tensor_id() function in tensorflow::__anonfdc77a490111::SymbolPredicate
316 explicit IntSymbolPredicate(int64_t id, TensorId tensor_id, in IntSymbolPredicate()
338 TensorId tensor_id() const { return tensor_id_; } in tensor_id() function in tensorflow::__anonfdc77a490111::IntSymbolPredicate
418 TensorId tensor_id(node->name(), output_idx); in MakeSymbolPredicate() local
454 TensorId tensor_id(node->name(), output_idx); in MakeSymbolPredicate() local
1558 for (TensorId tensor_id : tensor_ids) { in Print() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/ |
| H A D | tensorflow_profiler_logger.cc | 52 int64_t tensor_id, const std::string& name, in AddTraceMeInternal()
87 int64_t tensor_id = reinterpret_cast<int64_t>(tensor->data.raw); in AddTraceMe() local
177 int64_t tensor_id = arena_id; in AddArenaTrace() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | tensor_id_test.cc | 96 TensorId tensor_id = ParseTensorName(input); in TEST() local
110 TensorId tensor_id = ParseTensorName(input); in TEST() local
|
| H A D | tensor_id.cc | 65 bool IsTensorIdControl(const TensorId& tensor_id) { in IsTensorIdControl()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| H A D | utils.cc | 32 bool TensorTypeMatch(int tensor_id, TfLiteContext* context, in TensorTypeMatch()
124 int tensor_id; in IsNodeSupportedByHexagon() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| H A D | inference_context.cc | 560 for (auto& tensor_id : node_tensor_ids) { in AllocateBufferBasedTensors() local
642 for (auto& tensor_id : node_tensor_ids) { in AllocateStrongShapesTensors() local
699 absl::Status InferenceContext::SetTensor(const ValueId& tensor_id, in SetTensor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | tpu_embedding_software_deduplication_rewrite_pass.cc | 700 const TensorId tensor_id = ParseTensorName(input); in GetControlInputs() local
717 const TensorId tensor_id = ParseTensorName(input); in GetDataInputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| H A D | export_graphdef.cc | 477 auto tensor_id = ParseTensorName(name); in Convert() local
495 auto tensor_id = ParseTensorName(name); in Convert() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | object_reader.h | 71 const int32_t tensor_id = node_->inputs->data[index]; in ReadTensor() local
|