| /external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/ |
| decode_csv_vectorizer.cc | 16 #include "tensorflow/core/graph/node_builder.h"
40 auto node_builder = NodeBuilder(node.type_string(), node.type_string()) variable
45 node_builder = node_builder.Attr(attr.first, attr.second);
47 TF_RETURN_IF_ERROR(node_builder.Finalize(outer_scope, &new_node));
|
| parse_single_example_vectorizer.cc | 18 #include "tensorflow/core/graph/node_builder.h"
72 auto node_builder = variable
84 node_builder = node_builder.Attr(attr, *val);
87 TF_RETURN_IF_ERROR(node_builder.Finalize(outer_scope, &new_node));
|
| cwise_op_vectorizer.cc | 21 #include "tensorflow/core/graph/node_builder.h"
120 auto node_builder = NodeBuilder(strings::StrCat("vectorized/", node.name()), local
123 node_builder = node_builder.Input(input.node, input.output_index);
126 node_builder = node_builder.Attr(attr_slice.first, attr_slice.second);
128 TF_RETURN_IF_ERROR(node_builder.Finalize(outer_scope, &new_node));
|
| /external/tensorflow/tensorflow/c/ |
| c_api_internal.h | 38 #include "tensorflow/core/graph/node_builder.h"
117 : node_builder(node_name, op_type, g->graph.op_registry()), graph(g) {}
119 tensorflow::NodeBuilder node_builder; member in struct:TF_OperationDescription
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| vectorization_utils_test.cc | 34 #include "tensorflow/core/graph/node_builder.h"
895 auto node_builder = NodeBuilder("op", op_type); local
[all...] |
| vectorization_utils.cc | 32 #include "tensorflow/core/graph/node_builder.h"
429 auto node_builder = [](StringPiece op) { local
433 auto make_const = [&node_builder](const Input::Initializer& val, Graph* graph,
436 return node_builder("Const")
449 node_builder("Shape").Input(input_node).Finalize(g, &shape_node));
457 TF_RETURN_IF_ERROR(node_builder("StridedSlice")
465 TF_RETURN_IF_ERROR(node_builder("Reshape")
472 TF_RETURN_IF_ERROR(node_builder("Shape")
478 node_builder("OnesLike").Input(ones_shape).Finalize(g, &ones));
484 TF_RETURN_IF_ERROR(node_builder("Concat"
[all...] |
