Searched refs:op_node (Results 1 – 4 of 4) sorted by relevance
| /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 (size_t 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/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 | 42 Status Unary(const FDH::Node& op_node, const Tensor& x, const DataType dst, in Unary() argument
52 op_node, in Unary()
97 const FDH::Node op_node = {{"y"}, op, {"x"}, {{"T", x.dtype()}}}; in Unary() local
98 return Unary(op_node, x, x.dtype(), y); in Unary()
110 const FDH::Node op_node = { in SymCastGrad() local
112 TF_CHECK_OK(Unary(op_node, x, dst, &ret)); in SymCastGrad()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 555 Status EvaluateBoolOpForConstantOperands(const NodeDef& op_node, in EvaluateBoolOpForConstantOperands() argument
574 EvaluateNode(op_node, inputs, cpu_device, resource_mgr, &outputs)); in EvaluateBoolOpForConstantOperands()
|