Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/core/framework/
Dnode_def_util.h198 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
200 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
202 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
204 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
206 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
208 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
210 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
212 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
214 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
216 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name,
[all …]
Dnode_def_util.cc209 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
219 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, \
378 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in GetNodeAttr() function
389 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in GetNodeAttr() function
412 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in GetNodeAttr() function
435 Status GetNodeAttr(const AttrSlice& attrs, StringPiece attr_name, in GetNodeAttr() function
438 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, attr_name, &str_value)); in GetNodeAttr()
452 GetNodeAttr(node_or_attrs, arg_def.number_attr(), &repeats)); in AddArgToSig()
465 GetNodeAttr(node_or_attrs, arg_def.type_attr(), &dtype)); in AddArgToSig()
681 return GetNodeAttr(attrs, arg_def.number_attr(), num); in ComputeArgRange()
[all …]
/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.cc45 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "N", &N)); in PackGrad()
47 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "axis", &axis)); in PackGrad()
62 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "axis", &axis)); in UnpackGrad()
100 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "axis", &axis)); in QuantizeAndDequantizeV4GradHelper()
203 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "message", &message)); in CheckNumericsGrad()
256 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "batch_dim", &batch_dim)); in ReverseSequenceGrad()
258 TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "seq_dim", &seq_dim)); in ReverseSequenceGrad()
326 GetNodeAttr(op.node()->attrs(), "block_size", &block_size)); in SpaceToBatchGrad()
350 GetNodeAttr(op.node()->attrs(), "block_size", &block_size)); in BatchToSpaceGrad()
374 GetNodeAttr(op.node()->attrs(), "block_size", &block_size)); in SpaceToDepthGrad()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc1047 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &value)); in GetTensorListLength()
1050 TF_CHECK_OK(GetNodeAttr(n->def(), attr_name, &N)); in GetTensorListLength()
1124 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetConv2DOrBiasAdd()
1169 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetPadOrConv2D()
1205 TF_CHECK_OK(GetNodeAttr(conv_node->def(), "padding", &padding)); in GetPadOrConv2D()
1254 TF_CHECK_OK(GetNodeAttr(conv_node->def(), "padding", &padding)); in GetPadOrFusedConv2D()
1292 TF_CHECK_OK(GetNodeAttr(m->def(), "T", &T_m)); in GetConv2DBackpropFilterOrBiasAddGrad()
1340 GetNodeAttr(conv2d_backprop_filter->def(), "padding", &padding)); in GetConv2DBackpropFilterOrBiasAddGrad()
1415 TF_CHECK_OK(GetNodeAttr(perm_node->def(), "value", &proto)); in CheckForTranspose()
1418 TF_CHECK_OK(GetNodeAttr(perm_node->def(), "dtype", &type)); in CheckForTranspose()
[all …]
Dimmutable_executor_state.cc115 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "send_device", &send_device)); in Initialize()
116 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "recv_device", &recv_device)); in Initialize()
161 GetNodeAttr(n->attrs(), "is_constant", &is_constant_enter)); in Initialize()
165 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "frame_name", &frame_name)); in Initialize()
170 GetNodeAttr(n->attrs(), "parallel_iterations", &parallel_iterations)); in Initialize()
216 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "frame_name", &enter_name)); in Initialize()
313 GetNodeAttr(curr_node->attrs(), "frame_name", &frame_name)); in BuildControlFlowInfo()
Dmkl_tfconversion_pass.cc182 if (GetNodeAttr(src->def(), "data_format", &data_format) == Status::OK() && in InsertConversionNodeOnEdge()
267 if (GetNodeAttr(edges[0]->src()->def(), "data_format", &data_format) == in InsertInputConversionNode()
340 (GetNodeAttr(src->def(), "T", &src_datatype) == Status::OK() && in RunPass()
343 (GetNodeAttr(dst->def(), "T", &dst_datatype) == Status::OK() && in RunPass()
381 if (!n->IsOp() || (GetNodeAttr(n->def(), "T", &datatype) != Status::OK())) { in RunPass()
389 if ((GetNodeAttr(n->def(), "T", &input_datatype) == Status::OK()) && in RunPass()
Daccumulate_n_optimizer.cc127 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in RewriteIntoTempVariable()
129 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "shape", &shape)); in RewriteIntoTempVariable()
237 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "T", &dtype)); in RewriteIntoAddN()
239 TF_RETURN_IF_ERROR(GetNodeAttr(n_attrs, "N", &num_inputs)); in RewriteIntoAddN()
Dkernel_benchmark_testlib.cc149 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device", &send_device)); in GetRendezvousKey()
151 TF_CHECK_OK(GetNodeAttr(node->attrs(), "recv_device", &recv_device)); in GetRendezvousKey()
153 TF_CHECK_OK(GetNodeAttr(node->attrs(), "tensor_name", &tensor_name)); in GetRendezvousKey()
155 TF_CHECK_OK(GetNodeAttr(node->attrs(), "send_device_incarnation", in GetRendezvousKey()
Dshape_refiner.cc79 TF_RETURN_IF_ERROR(GetNodeAttr(AttrSlice(node->def()), "index", &index)); in InferShapesForFunctionSubNode()
107 TF_RETURN_IF_ERROR(GetNodeAttr(AttrSlice(node->def()), "index", &index)); in InferShapesForFunctionSubNode()
588 GetNodeAttr(slice_node->attrs(), "begin_mask", &begin_mask)); in PartialStridedSliceShape()
589 TF_RETURN_IF_ERROR(GetNodeAttr(slice_node->attrs(), "end_mask", &end_mask)); in PartialStridedSliceShape()
591 GetNodeAttr(slice_node->attrs(), "ellipsis_mask", &ellipsis_mask)); in PartialStridedSliceShape()
593 GetNodeAttr(slice_node->attrs(), "new_axis_mask", &new_axis_mask)); in PartialStridedSliceShape()
595 GetNodeAttr(slice_node->attrs(), "shrink_axis_mask", &shrink_axis_mask)); in PartialStridedSliceShape()
668 if (!GetNodeAttr(AttrSlice(node->def()), "_disable_call_shape_inference", in RunShapeFn()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dside_effect_util.cc54 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), attr_name, &branch_func)); in SetDeviceOrdinalAttributeForNode()
64 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), attr_name, &branch_func)); in SetDeviceOrdinalAttributeForNode()
86 if (!GetNodeAttr(n->attrs(), kXlaTokenInputNodesAttrName, in CalculateTokenInputsForOutputToken()
99 TF_CHECK_OK(GetNodeAttr(n->def(), in CalculateTokenInputsForOutputToken()
116 if (GetNodeAttr(n->attrs(), kXlaTokenInputNodesAttrName, &token_input_nodes) in HasSideEffectingNodes()
Drearrange_function_argument.cc186 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "T", &t)); in CalculateRetvalRearrange()
203 TF_RETURN_IF_ERROR(GetNodeAttr(arg->def(), "index", &src_index)); in CalculateRetvalRearrange()
282 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "T", &types)); in MaybeRewriteWhileNode()
309 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), attr_name, &attr_value)); in MaybeRewriteWhileNode()
323 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "T", &dtype)); in MaybeRewriteWhileNode()
335 TF_RETURN_IF_ERROR(GetNodeAttr(input_node->def(), "index", &index)); in MaybeRewriteWhileNode()
389 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "Tin", &in_types)); in MaybeRewriteIfNode()
396 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "Tout", &out_types)); in MaybeRewriteIfNode()
435 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), attr_name, &f)); in MaybeRewriteIfNode()
524 TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "f", &func_name_attrs)); in RearrangeFunctionArguments()
Dgraph_compiler_util.cc175 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), kFeedIdAttr, &feed_id)); in RewriteAndPruneGraph()
182 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), kFetchIdAttr, &fetch_id)); in RewriteAndPruneGraph()
206 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in CollectArgNodes()
239 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), "T", &arg.type)); in CreateXlaArgs()
241 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), kShapeAttr, &shape)); in CreateXlaArgs()
243 TF_RETURN_IF_ERROR(GetNodeAttr(node->attrs(), kDebugNameAttr, &arg.name)); in CreateXlaArgs()
/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()
555 TF_RETURN_IF_ERROR(GetNodeAttr(attrs, "Tidx", &itype)); in BroadcastToGrad()
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/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc61 TF_RETURN_IF_ERROR(GetNodeAttr(n.attrs(), "index", index)); in GetIndexAttr()
90 if (!ret || !GetNodeAttr(arg_source_tensors.at(index).node->attrs(), in RewriteSubgraph()
108 TF_CHECK_OK(GetNodeAttr(arg_source_tensors.at(index).node->attrs(), "index", in RewriteSubgraph()
116 if (!GetNodeAttr(n.attrs(), "_is_guaranteed_constant", &guaranteed_constant) in RewriteSubgraph()
164 GetNodeAttr(metadata_node->attrs(), "computation_shape", &shape)); in RewriteSubgraph()
337 TF_RETURN_IF_ERROR(GetNodeAttr(xla_node->def(), in UpdateMirroredVariableIndices()
453 TF_RETURN_IF_ERROR(GetNodeAttr(xla_node->attrs(), "Tinputs", &input_types)); in MoveHeadOutsideCompilationToHost()
455 TF_RETURN_IF_ERROR(GetNodeAttr(xla_node->attrs(), "num_distributed_variables", in MoveHeadOutsideCompilationToHost()
459 GetNodeAttr(xla_node->attrs(), "num_replicas", &num_replicas)); in MoveHeadOutsideCompilationToHost()
484 TF_RETURN_IF_ERROR(GetNodeAttr(e->src()->attrs(), "index", &index)); in MoveHeadOutsideCompilationToHost()
[all …]
Dvariable_merger_pass.cc60 GetNodeAttr(nodes[i]->attrs(), "container", &containers[i])); in MergeVarHandleOps()
62 GetNodeAttr(nodes[i]->attrs(), "shared_name", &names[i])); in MergeVarHandleOps()
63 TF_RETURN_IF_ERROR(GetNodeAttr(nodes[i]->attrs(), "dtype", &dtypes[i])); in MergeVarHandleOps()
64 TF_RETURN_IF_ERROR(GetNodeAttr(nodes[i]->attrs(), "shape", &shapes[i])); in MergeVarHandleOps()
103 TF_RETURN_IF_ERROR(GetNodeAttr(nodes[i]->attrs(), "dtype", &dtypes[i])); in MergeReadVariableOps()
/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()
/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/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/core/tpu/ops/
Dtpu_compile_op.cc38 GetNodeAttr(c->attrs(), "num_computations", &num_computations)); in __anondb58f4d00102()
98 GetNodeAttr(c->attrs(), "num_computations", &num_computations)); in __anondb58f4d00302()
/external/tensorflow/tensorflow/compiler/jit/
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()
316 TF_CHECK_OK(GetNodeAttr(host_compute_0->attrs(), "tpu_core", &tpu_core)); in TEST_F()
318 TF_CHECK_OK(GetNodeAttr(host_compute_1->attrs(), "tpu_core", &tpu_core)); in TEST_F()
323 TF_CHECK_OK(GetNodeAttr(host_compute_0->attrs(), "shapes", &shapes)); in TEST_F()
325 TF_CHECK_OK(GetNodeAttr(host_compute_1->attrs(), "shapes", &shapes)); in TEST_F()
331 TF_CHECK_OK(GetNodeAttr(host_compute_0->attrs(), "shape_inference_graph", in TEST_F()
334 TF_CHECK_OK(GetNodeAttr(host_compute_1->attrs(), "shape_inference_graph", in TEST_F()
[all …]
Dextract_outside_compilation_pass.cc93 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in GetArgDataTypes()
95 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "T", &dtype)); in GetArgDataTypes()
145 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in ReplaceArgNodesWithRecvAtHostNode()
194 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in GetRetDataTypes()
196 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "T", &dtype)); in GetRetDataTypes()
228 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in BuildSendFromHostNode()
262 TF_RETURN_IF_ERROR(GetNodeAttr(n->attrs(), "index", &index)); in ReplaceRetNodesWithSendFromHostNode()
289 if (!GetNodeAttr(e->src()->attrs(), kXlaInferredShapesAttrName, &shapes) in GetInferredInputShapes()
314 TF_RETURN_IF_ERROR(GetNodeAttr( in BuildXlaHostComputeNodeDef()
358 TF_RETURN_IF_ERROR(GetNodeAttr(call_node->attrs(), "Tinputs", &input_dtypes)); in BuildXlaHostComputeNodeDef()
[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