Home
last modified time | relevance | path

Searched refs:GetNodeAttr (Results 1 – 25 of 107) sorted by relevance

12345

/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.h181 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
183 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
185 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
187 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
189 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
191 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
193 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
195 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
197 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
199 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
[all …]
Dgraph_to_functiondef.cc128 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &type)); in GraphToFunctionDef()
129 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "index", &index)); in GraphToFunctionDef()
145 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &type)); in GraphToFunctionDef()
146 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "index", &index)); in GraphToFunctionDef()
Dnode_def_util.cc211 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
221 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
316 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in GetNodeAttr() function
327 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in GetNodeAttr() function
336 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in GetNodeAttr() function
353 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, arg_def.number_attr(), &repeats)); in AddArgToSig()
361 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, arg_def.type_attr(), &dtype)); in AddArgToSig()
562 return GetNodeAttr(node_def, arg_def.number_attr(), num); in ComputeArgRange()
779 TF_RETURN_IF_ERROR(GetNodeAttr(*node_def, "frame_name", &frame_name)); in ADD_ATTR()
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc788 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &value)); in GetTensorListLength()
791 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &N)); in GetTensorListLength()
865 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetConv2DOrBiasAdd()
905 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetPadOrConv2D()
936 TF_CHECK_OK(GetNodeAttr(conv_node->def(), "padding", &padding)); in GetPadOrConv2D()
985 TF_CHECK_OK(GetNodeAttr(conv_node->def(), "padding", &padding)); in GetPadOrFusedConv2D()
1022 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetConv2DBackpropFilterOrBiasAddGrad()
1122 TF_CHECK_OK(GetNodeAttr(perm_node->def(), "value", &proto)); in CheckForTranspose()
1125 GetNodeAttr(perm_node->def(), "dtype", &type); in CheckForTranspose()
1174 TF_CHECK_OK(GetNodeAttr(node->def(), "T", &T)); in CheckForMklOp()
[all …]
Dmkl_tfconversion_pass.cc181 if (GetNodeAttr(src->def(), "data_format", &data_format) == Status::OK() && in InsertConversionNodeOnEdge()
265 if (GetNodeAttr(edges[0]->src()->def(), "data_format", &data_format) == in InsertInputConversionNode()
337 (GetNodeAttr(src->def(), "T", &src_datatype) == Status::OK() && in RunPass()
340 (GetNodeAttr(dst->def(), "T", &dst_datatype) == Status::OK() && in RunPass()
378 if (!n->IsOp() || (GetNodeAttr(n->def(), "T", &datatype) != Status::OK())) { in RunPass()
386 if ((GetNodeAttr(n->def(), "T", &input_datatype) == Status::OK()) && in RunPass()
/external/tensorflow/tensorflow/cc/gradients/
Dnn_grad.cc150 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "alpha", &alpha)); in LeakyReluGradHelper()
163 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "alpha", &alpha)); in LeakyReluGradGradHelper()
204 GetNodeAttr(op.output(0).node()->attrs(), "data_format", &data_format)); in BiasAddGradHelper()
221 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "data_format", &data_format)); in Conv2DGrad()
222 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "padding", &padding)); in Conv2DGrad()
223 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "strides", &strides)); in Conv2DGrad()
224 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "use_cudnn_on_gpu", &use_cudnn_on_gpu)); in Conv2DGrad()
248 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "data_format", &data_format)); in MaxPoolGradHelper()
249 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "ksize", &ksize)); in MaxPoolGradHelper()
250 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "padding", &padding)); in MaxPoolGradHelper()
[all …]
Dimage_grad.cc31 GetNodeAttr(op.node()->attrs(), "align_corners", &align_corners)); in ResizeNearestNeighborGradHelper()
33 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "half_pixel_centers", in ResizeNearestNeighborGradHelper()
53 GetNodeAttr(op.node()->attrs(), "align_corners", &align_corners)); in ResizeBilinearGradHelper()
55 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "half_pixel_centers", in ResizeBilinearGradHelper()
71 GetNodeAttr(op.node()->attrs(), "align_corners", &align_corners)); in ResizeBicubicGradHelper()
73 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "half_pixel_centers", in ResizeBicubicGradHelper()
90 GetNodeAttr(op.node()->attrs(), "kernel_type", &kernel_type)); in ScaleAndTranslateGradHelper()
92 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "antialias", &antialias)); in ScaleAndTranslateGradHelper()
111 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "method", &method)); in CropAndResizeGradHelper()
112 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "T", &input_type)); in CropAndResizeGradHelper()
Darray_grad.cc46 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "N", &N)); in PackGrad()
48 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "axis", &axis)); in PackGrad()
63 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "axis", &axis)); in UnpackGrad()
194 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "message", &message)); in CheckNumericsGrad()
247 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "batch_dim", &batch_dim)); in ReverseSequenceGrad()
249 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "seq_dim", &seq_dim)); in ReverseSequenceGrad()
317 GetNodeAttr(op.node()->attrs(), "block_size", &block_size)); in SpaceToBatchGrad()
341 GetNodeAttr(op.node()->attrs(), "block_size", &block_size)); in BatchToSpaceGrad()
365 GetNodeAttr(op.node()->attrs(), "block_size", &block_size)); in SpaceToDepthGrad()
376 GetNodeAttr(op.node()->attrs(), "block_size", &block_size)); in DepthToSpaceGrad()
[all …]
/external/tensorflow/tensorflow/core/ops/
Darray_grad.cc145 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "N", &N)); in ConcatGradHelper()
147 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "T", &T)); in ConcatGradHelper()
270 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "N", &N)); in ArrayToListGrad()
417 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "Tidx", &itype)); in ReverseV2Grad()
441 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "Index", &itype)); in SliceGrad()
476 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "Index", &itype)); in StridedSliceGrad()
513 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "Index", &itype)); in StridedSliceGradGrad()
Dfunctional_grad.cc26 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "f", &func)); in MapAccumulateGrad()
28 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "T", &T)); in MapAccumulateGrad()
30 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "K", &k)); in MapAccumulateGrad()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.cc46 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), attr_name, &branch_func)); in SetDeviceOrdinalAttributeForNode()
56 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), attr_name, &branch_func)); in SetDeviceOrdinalAttributeForNode()
78 if (!GetNodeAttr(n->attrs(), kXlaTokenInputNodesAttrName, in CalculateTokenInputsForOutputToken()
104 if (GetNodeAttr(n->attrs(), kXlaTokenInputNodesAttrName, &token_input_nodes) in HasSideEffectingNodes()
Dtf2xla.cc182 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), kFeedIdAttr, &feed_id)); in RewriteAndPruneGraph()
189 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), kFetchIdAttr, &fetch_id)); in RewriteAndPruneGraph()
213 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in CollectArgNodes()
243 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &arg.type)); in CreateXlaArgs()
245 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), kShapeAttr, &shape)); in CreateXlaArgs()
247 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), kDebugNameAttr, &arg.name)); in CreateXlaArgs()
Dconst_analysis.cc66 status = GetNodeAttr(node->attrs(), "index", &index); in BackwardsConstAnalysis()
139 TF_RETURN_IF_ERROR(GetNodeAttr(node->def(), "cond", &cond_function)); in GetCompileTimeConstInputs()
141 TF_RETURN_IF_ERROR(GetNodeAttr(node->def(), "body", &body_function)); in GetCompileTimeConstInputs()
/external/tensorflow/tensorflow/core/common_runtime/
Dparallel_concat_optimizer.cc59 if (GetNodeAttr(n_attrs, "_class", &colo).ok()) { in Run()
69 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in Run()
71 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "shape", &shape)); in Run()
Dkernel_benchmark_testlib.cc119 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device", &send_device)); in GetRendezvousKey()
121 TF_CHECK_OK(GetNodeAttr(node->attrs(), "recv_device", &recv_device)); in GetRendezvousKey()
123 TF_CHECK_OK(GetNodeAttr(node->attrs(), "tensor_name", &tensor_name)); in GetRendezvousKey()
125 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device_incarnation", in GetRendezvousKey()
Daccumulate_n_optimizer.cc84 if (GetNodeAttr(n_attrs, kColocationAttrName, &colo).ok()) { in rewriteNode()
95 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in rewriteNode()
97 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "shape", &shape)); in rewriteNode()
/external/tensorflow/tensorflow/cc/ops/
Dconst_op_test.cc31 TF_EXPECT_OK(GetNodeAttr(n->attrs(), "value", &tensor)); in ExpectNodeEqual()
33 TF_EXPECT_OK(GetNodeAttr(n->attrs(), "dtype", &dtype)); in ExpectNodeEqual()
42 TF_EXPECT_OK(GetNodeAttr(n->attrs(), "value", &tensor)); in ExpectTypeAndShape()
44 TF_EXPECT_OK(GetNodeAttr(n->attrs(), "dtype", &dtype)); in ExpectTypeAndShape()
/external/tensorflow/tensorflow/core/util/
Dmirror_pad_mode.cc24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name, in GetNodeAttr() function
27 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, attr_name, &str_value)); in GetNodeAttr()
Dpadding.cc24 Status GetNodeAttr(const NodeDef& node_def, StringPiece attr_name, in GetNodeAttr() function
27 TF_RETURN_IF_ERROR(GetNodeAttr(node_def, attr_name, &str_value)); in GetNodeAttr()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Dunpack_vectorizer.cc34 TF_RETURN_IF_ERROR(GetNodeAttr(node.attrs(), "axis", &axis)); in Vectorize()
45 TF_RETURN_IF_ERROR(GetNodeAttr(node.attrs(), "num", &num)); in Vectorize()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc47 Status s = GetNodeAttr(n->attrs(), attr_name, &attr_value); in AppendToListAttr()
154 TF_RETURN_IF_ERROR(GetNodeAttr(dst->attrs(), in PreprocessDataEdgesBetweenOutsideCompilations()
209 TF_RETURN_IF_ERROR(GetNodeAttr( in PostprocessDataEdgesBetweenOutsideCompilations()
211 TF_RETURN_IF_ERROR(GetNodeAttr( in PostprocessDataEdgesBetweenOutsideCompilations()
277 GetNodeAttr(n->attrs(), kXlaControlDependenciesWithinXlaClusterAttrName, in PostprocessControlEdgesBetweenOutsideCompilations()
Dextract_outside_compilation_pass.cc80 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in GetArgDataTypes()
82 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "T", &dtype)); in GetArgDataTypes()
132 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in ReplaceArgNodesWithRecvAtHostNode()
181 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in GetRetDataTypes()
183 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "T", &dtype)); in GetRetDataTypes()
215 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in BuildSendFromHostNode()
248 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in ReplaceRetNodesWithSendFromHostNode()
275 if (!GetNodeAttr(e->src()->attrs(), kXlaInferredShapesAttrName, &shapes) in GetInferredInputShapes()
294 TF_RETURN_IF_ERROR(GetNodeAttr( in BuildXlaHostComputeNodeDef()
318 TF_RETURN_IF_ERROR(GetNodeAttr(call_node->attrs(), "Tinputs", &input_dtypes)); in BuildXlaHostComputeNodeDef()
[all …]
Dencapsulate_xla_computations_pass.cc47 if (!GetNodeAttr(n->attrs(), in IsCpuGpuCompile()
65 if (!GetNodeAttr(n.attrs(), "_is_guaranteed_constant", &guaranteed_constant) in is_guaranteed_constant()
74 TF_RETURN_IF_ERROR(GetNodeAttr(n.attrs(), "index", index)); in GetIndexAttr()
249 if (GetNodeAttr(n->attrs(), kXlaClusterAttr, &name).ok()) { in BuildXlaLaunchOps()
258 TF_RETURN_IF_ERROR(GetNodeAttr(launch->attrs(), "_variable_start_index", in BuildXlaLaunchOps()
Dextract_outside_compilation_pass_test.cc79 GetNodeAttr(recv_at_host->attrs(), "Toutputs", &recv_at_host_dtypes)); in TEST()
92 GetNodeAttr(send_from_host->attrs(), "Tinputs", &send_from_host_dtypes)); in TEST()
118 TF_CHECK_OK(GetNodeAttr(AttrSlice(&call_node_def.attr()), in TEST()
223 TF_CHECK_OK(GetNodeAttr(AttrSlice(&call_node_def.attr()), "shapes", &shapes)); in TEST()
320 TF_CHECK_OK(GetNodeAttr(host_compute_0->attrs(), "tpu_core", &tpu_core)); in TEST_F()
322 TF_CHECK_OK(GetNodeAttr(host_compute_1->attrs(), "tpu_core", &tpu_core)); in TEST_F()
327 TF_CHECK_OK(GetNodeAttr(host_compute_0->attrs(), "shapes", &shapes)); in TEST_F()
329 TF_CHECK_OK(GetNodeAttr(host_compute_1->attrs(), "shapes", &shapes)); in TEST_F()
335 TF_CHECK_OK(GetNodeAttr(host_compute_0->attrs(), "shape_inference_graph", in TEST_F()
338 TF_CHECK_OK(GetNodeAttr(host_compute_1->attrs(), "shape_inference_graph", in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc33 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "reduction", &reduction)); in ReplaceReduce()
35 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &dtype)); in ReplaceReduce()
119 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &dtype)); in ReplaceBroadcast()
185 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "shape", &shape_proto)); in ReplaceBroadcast()

12345