HomeSort by relevance Sort by last modified time
    Searched refs:op_node (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/core/kernels/fuzzing/
fuzz_session.h 40 auto op_node = \
42 (void)tensorflow::ops::opName(scope.WithOpName("output"), op_node); \
  /external/tensorflow/tensorflow/core/grappler/optimizers/data/
vectorization_utils.cc 176 Status AddConversionMapping(Node* op_node);
231 Status Vectorization::AddConversionMapping(Node* op_node) {
232 for (auto edge : op_node->in_edges()) {
240 auto vectorizer = VectorizerRegistry::Global()->Get(op_node->type_string());
243 op_node->type_string());
246 inputs.reserve(op_node->num_inputs());
247 outputs.reserve(op_node->num_outputs());
250 TF_RETURN_IF_ERROR(op_node->input_edges(&input_edges));
254 for (size_t i = 0; i < op_node->num_inputs(); ++i) {
275 Status s = vectorizer->Vectorize(*op_node, outer_scope_.get()
    [all...]
  /external/tensorflow/tensorflow/core/ops/
math_grad_test.cc 41 // dst is the output dtype of op_node.
42 Status Unary(const FDH::Node& op_node, const Tensor& x, const DataType dst,
52 op_node,
97 const FDH::Node op_node = {{"y"}, op, {"x"}, {{"T", x.dtype()}}}; local
98 return Unary(op_node, x, x.dtype(), y);
110 const FDH::Node op_node = { local
112 TF_CHECK_OK(Unary(op_node, x, dst, &ret));
    [all...]
  /external/tensorflow/tensorflow/core/grappler/optimizers/
loop_optimizer.cc 555 Status EvaluateBoolOpForConstantOperands(const NodeDef& op_node,
574 EvaluateNode(op_node, inputs, cpu_device, resource_mgr, &outputs));
    [all...]

Completed in 735 milliseconds