HomeSort by relevance Sort by last modified time
    Searched refs:GetNodeAttr (Results 1 - 25 of 72) sorted by null

1 2 3

  /external/tensorflow/tensorflow/core/util/
mirror_pad_mode.cc 24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name,
27 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, attr_name, &str_value));
padding.cc 24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name,
27 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, attr_name, &str_value));
mirror_pad_mode.h 47 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name,
padding.h 48 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name,
  /external/tensorflow/tensorflow/core/ops/
functional_grad.cc 26 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "f", &func));
28 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "T", &T));
30 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "K", &k));
array_grad.cc 145 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "N", &N));
147 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "T", &T));
249 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "N", &N));
375 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "Tidx", &itype));
399 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "Index", &itype));
434 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "Index", &itype));
471 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "Index", &itype));
  /external/tensorflow/tensorflow/core/framework/
node_def_util.h 168 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
170 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
172 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
174 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
176 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
178 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
180 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
182 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
184 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
186 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name
    [all...]
memory_types.cc 138 if (GetNodeAttr(ndef, "_input_hostmem", &hostmem_attr).ok()) {
145 if (GetNodeAttr(ndef, "_output_hostmem", &hostmem_attr).ok()) {
resource_op_kernel_test.cc 65 return GetNodeAttr(def(), "code", &(*resource)->code);
70 TF_RETURN_IF_ERROR(GetNodeAttr(def(), "code", &code));
fake_input.cc 104 Status status = GetNodeAttr(*node_def_, arg_->type_list_attr(), &dts);
131 Status status = GetNodeAttr(*node_def_, arg_->number_attr(), n);
148 Status status = GetNodeAttr(*node_def_, arg_->type_attr(), dt);
  /external/tensorflow/tensorflow/cc/gradients/
nn_grad.cc 112 GetNodeAttr(op.output(0).node()->attrs(), "data_format", &data_format));
129 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "data_format", &data_format));
130 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "padding", &padding));
131 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "strides", &strides));
132 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "use_cudnn_on_gpu", &use_cudnn_on_gpu));
159 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "data_format", &data_format));
160 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "ksize", &ksize));
161 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "padding", &padding));
162 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "strides", &strides));
181 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "data_format", &data_format))
    [all...]
array_grad.cc 46 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "N", &N));
48 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "axis", &axis));
63 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "axis", &axis));
176 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "message", &message));
229 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "batch_dim", &batch_dim));
231 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "seq_dim", &seq_dim));
299 GetNodeAttr(op.node()->attrs(), "block_size", &block_size));
323 GetNodeAttr(op.node()->attrs(), "block_size", &block_size));
347 GetNodeAttr(op.node()->attrs(), "block_size", &block_size));
358 GetNodeAttr(op.node()->attrs(), "block_size", &block_size))
    [all...]
data_flow_grad.cc 68 GetNodeAttr(op.node()->attrs(), "num_partitions", &num_partitions));
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_tfgraph_builder_test.cc 56 static const tensorflow::AttrValue &GetNodeAttr(const tensorflow::NodeDef &node,
79 auto dims_value = GetNodeAttr(node, "dims");
84 auto shape_value = GetNodeAttr(node, "_output_shapes");
103 EXPECT_EQ(GetNodeAttr(node, "value").s(), "123");
104 EXPECT_EQ(GetNodeAttr(node, "type").s(), "S32");
105 EXPECT_EQ(GetNodeAttr(node, "tf_op_name").s(), "x");
106 EXPECT_EQ(GetNodeAttr(node, "tf_op_type").s(), "y");
  /external/tensorflow/tensorflow/core/graph/
mkl_layout_pass.cc 528 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &value));
531 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &N));
612 CHECK_EQ(GetNodeAttr(n->def(), "ksize", &ksize).ok(), true);
613 CHECK_EQ(GetNodeAttr(n->def(), "strides", &strides).ok(), true);
614 CHECK_EQ(GetNodeAttr(n->def(), "data_format", &data_format_str).ok(), true);
632 CHECK_EQ(GetNodeAttr(n->def(), "N", &num).ok(), true);
    [all...]
mkl_tfconversion_pass.cc 154 TF_CHECK_OK(GetNodeAttr(src->def(), "T", &src_datatype));
156 GetNodeAttr(dst->def(), "T", &dst_datatype) == Status::OK();
179 if (GetNodeAttr(src->def(), "data_format", &data_format) == Status::OK()) {
259 if (GetNodeAttr(edges[0]->src()->def(), "data_format", &data_format) ==
329 (GetNodeAttr(src->def(), "T", &src_datatype) == Status::OK() &&
332 (GetNodeAttr(dst->def(), "T", &dst_datatype) == Status::OK() &&
370 if (!n->IsOp() || (GetNodeAttr(n->def(), "T", &datatype) != Status::OK())) {
378 if ((GetNodeAttr(n->def(), "T", &input_datatype) == Status::OK()) &&
  /external/tensorflow/tensorflow/core/common_runtime/
parallel_concat_optimizer.cc 58 if (GetNodeAttr(n_attrs, "_class", &colo).ok()) {
68 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype));
70 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "shape", &shape));
accumulate_n_optimizer.cc 84 if (GetNodeAttr(n_attrs, kColocationAttrName, &colo).ok()) {
95 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype));
97 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "shape", &shape));
kernel_benchmark_testlib.cc 114 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device", &send_device));
116 TF_CHECK_OK(GetNodeAttr(node->attrs(), "recv_device", &recv_device));
118 TF_CHECK_OK(GetNodeAttr(node->attrs(), "tensor_name", &tensor_name));
120 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device_incarnation",
  /external/tensorflow/tensorflow/cc/ops/
const_op_test.cc 31 TF_EXPECT_OK(GetNodeAttr(n->attrs(), "value", &tensor));
33 TF_EXPECT_OK(GetNodeAttr(n->attrs(), "dtype", &dtype));
42 TF_EXPECT_OK(GetNodeAttr(n->attrs(), "value", &tensor));
44 TF_EXPECT_OK(GetNodeAttr(n->attrs(), "dtype", &dtype));
  /external/tensorflow/tensorflow/compiler/tf2xla/
const_analysis.cc 53 status = GetNodeAttr(node->attrs(), "index", &index);
tf2xla.cc 182 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), kFeedIdAttr, &feed_id));
189 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), kFetchIdAttr, &fetch_id));
213 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index));
243 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &arg.type));
244 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), kShapeAttr, &arg.shape));
245 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), kDebugNameAttr, &arg.name));
  /external/tensorflow/tensorflow/compiler/jit/
build_xla_launch_ops_pass.cc 64 GetNodeAttr(node->attrs(), kXlaNumConstantArgsAttr, &num_constant_args));
66 GetNodeAttr(node->attrs(), kXlaNumResourceArgsAttr, &num_resource_args));
graph_to_functiondef.cc 127 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &type));
128 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "index", &index));
144 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &type));
145 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "index", &index));
  /external/tensorflow/tensorflow/contrib/nccl/kernels/
nccl_rewrite.cc 31 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "reduction", &reduction));
33 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &dtype));
117 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &dtype));
183 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "shape", &shape_proto));

Completed in 211 milliseconds

1 2 3