| /external/tensorflow/tensorflow/core/kernels/ |
| map_stage_op.cc | 495 TF_RETURN_IF_ERROR(GetNodeAttr(ndef, "dtypes", &dtypes));
496 TF_RETURN_IF_ERROR(GetNodeAttr(ndef, "capacity", &capacity));
497 TF_RETURN_IF_ERROR(GetNodeAttr(ndef, "memory_limit", &memory_limit));
|
| padding_fifo_queue.cc | 271 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, "shapes", &requested_shapes));
|
| remote_fused_graph_execute_utils.cc | 403 status = GetNodeAttr(attrs, ATTR_OUTPUT_DATA_TYPES, data_types);
409 status = GetNodeAttr(attrs, ATTR_OUTPUT_SHAPES, shapes);
[all...] |
| sparse_conditional_accumulator.h | 365 ctx, GetNodeAttr(ctx->op_kernel().def(), "has_known_shape",
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| master_session.cc | 336 TF_CHECK_OK(GetNodeAttr(ndef, "client_terminated", &client_terminated));
339 TF_CHECK_OK(GetNodeAttr(ndef, "tensor_name", &name));
341 TF_CHECK_OK(GetNodeAttr(ndef, "send_device", &send_device));
343 TF_CHECK_OK(GetNodeAttr(ndef, "recv_device", &recv_device));
346 GetNodeAttr(ndef, "send_device_incarnation",
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| function.cc | 571 TF_CHECK_OK(GetNodeAttr(*x, "index", &xi));
573 TF_CHECK_OK(GetNodeAttr(*y, "index", &yi));
582 if (!GetNodeAttr(n, "T", &dt).ok()) {
[all...] |
| shape_inference.h | 784 return GetNodeAttr(*node_def_, attr_name, value);
|
| op_kernel.h | [all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| shape_refiner.cc | 76 TF_RETURN_IF_ERROR(GetNodeAttr(AttrSlice(node->def()), "index", &index));
96 TF_RETURN_IF_ERROR(GetNodeAttr(AttrSlice(node->def()), "index", &index));
[all...] |
| function.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| encapsulate_subgraphs_pass.cc | [all...] |
| mark_for_compilation_pass_test.cc | 63 if (GetNodeAttr(node->attrs(), kXlaClusterAttr, &cluster).ok()) {
|
| encapsulate_subgraphs_pass_test.cc | 680 if (!GetNodeAttr(n.attrs(), "_is_guaranteed_constant",
[all...] |
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| graph_transferer.cc | 487 TF_CHECK_OK(GetNodeAttr(node.attrs(), "value", &proto));
731 TF_CHECK_OK(GetNodeAttr(input_node->attrs(), "value", &tensor_proto));
742 TF_CHECK_OK(GetNodeAttr(input_node->attrs(), "value", &proto));
[all...] |
| /external/tensorflow/tensorflow/cc/framework/ |
| scope.cc | 218 if (GetNodeAttr(attrs, kColocationAttrName, &node_constraints).ok()) {
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| math_grad.cc | [all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph_constructor.cc | 564 if (!GetNodeAttr(node->attrs(), kAttrName, &shape_attrs).ok()) {
808 GetNodeAttr(node->attrs(), kColocationAttrName, &coloc_values);
[all...] |
| graph_partition_test.cc | 511 TF_CHECK_OK(GetNodeAttr(ndef, "send_device_incarnation", &val));
|
| graph_constructor_test.cc | 148 Status s = GetNodeAttr(n->attrs(), kColocationAttrName, &value);
997 s = GetNodeAttr(a->attrs(), "default_int", &value);
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_compiler.cc | 501 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index));
|
| functionalize_control_flow.cc | [all...] |
