Searched refs:op_node (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | op_builder.h | 295 const OpNode* op_node = builders_[i]->Build(); in Build() local
297 graph_id_, op_node->node_id, op_node->op_type, op_node->padding_type, in Build()
298 op_node->inputs.data(), op_node->inputs.size(), in Build()
299 op_node->outputs.data(), op_node->outputs.size()); in Build()
301 printf("Error adding node: id:%d, op_type:%d\n", op_node->node_id, in Build()
302 op_node->op_type); in Build()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils.cc | 176 Status AddConversionMapping(Node* op_node);
231 Status Vectorization::AddConversionMapping(Node* op_node) { in AddConversionMapping() argument
232 for (auto edge : op_node->in_edges()) { in AddConversionMapping()
240 auto vectorizer = VectorizerRegistry::Global()->Get(op_node->type_string()); in AddConversionMapping()
243 op_node->type_string()); in AddConversionMapping()
246 inputs.reserve(op_node->num_inputs()); in AddConversionMapping()
247 outputs.reserve(op_node->num_outputs()); in AddConversionMapping()
250 TF_RETURN_IF_ERROR(op_node->input_edges(&input_edges)); in AddConversionMapping()
254 for (int i = 0; i < op_node->num_inputs(); ++i) { in AddConversionMapping()
275 Status s = vectorizer->Vectorize(*op_node, outer_scope_.get(), in AddConversionMapping()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | remove_noop.cc | 54 Node* op_node = sequence.back(); in ApplyToNodesSequence() local
55 if (!remove_predicate_(graph, op_node)) { in ApplyToNodesSequence()
58 absl::Status status = RemoveFollowingNode(graph, op_node, prev_op_node); in ApplyToNodesSequence()
|
| D | merge_padding_with.cc | 56 Node* op_node = sequence.back(); in ApplyToNodesSequence() local
70 Attr* node_attr = absl::any_cast<Attr>(&op_node->operation.attributes); in ApplyToNodesSequence()
71 absl::Status status = RemovePrecedingNode(graph, pad_node, op_node); in ApplyToNodesSequence()
|
| /external/tensorflow/tensorflow/security/fuzzing/op_fuzzing/ |
| D | fuzz_session.h | 39 auto op_node = \
41 (void)tensorflow::ops::opName(scope.WithOpName("output"), op_node); \
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | fuzz_session.h | 40 auto op_node = \
42 (void)tensorflow::ops::opName(scope.WithOpName("output"), op_node); \
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | math_grad_test.cc | 43 Status Unary(const FDH::Node& op_node, const Tensor& x, const DataType dst, in Unary() argument
53 op_node, in Unary()
98 const FDH::Node op_node = {{"y"}, op, {"x"}, {{"T", x.dtype()}}}; in Unary() local
99 return Unary(op_node, x, x.dtype(), y); in Unary()
111 const FDH::Node op_node = { in SymCastGrad() local
113 TF_CHECK_OK(Unary(op_node, x, dst, &ret)); in SymCastGrad()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 554 Status EvaluateBoolOpForConstantOperands(const NodeDef& op_node, in EvaluateBoolOpForConstantOperands() argument
560 VLOG(4) << "Evaluate bool op: op_node=" << op_node.name() in EvaluateBoolOpForConstantOperands()
576 EvaluateNode(op_node, inputs, cpu_device, resource_mgr, &outputs)); in EvaluateBoolOpForConstantOperands()
|