/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));
|