/external/tensorflow/tensorflow/core/framework/ |
D | node_def_util.h | 181 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 …]
|
D | graph_to_functiondef.cc | 128 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()
|
D | node_def_util.cc | 211 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/ |
D | mkl_layout_pass.cc | 788 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 …]
|
D | mkl_tfconversion_pass.cc | 181 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/ |
D | nn_grad.cc | 150 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 …]
|
D | image_grad.cc | 31 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()
|
D | array_grad.cc | 46 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/ |
D | array_grad.cc | 145 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()
|
D | functional_grad.cc | 26 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/ |
D | side_effect_util.cc | 46 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()
|
D | tf2xla.cc | 182 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()
|
D | const_analysis.cc | 66 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/ |
D | parallel_concat_optimizer.cc | 59 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()
|
D | kernel_benchmark_testlib.cc | 119 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()
|
D | accumulate_n_optimizer.cc | 84 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/ |
D | const_op_test.cc | 31 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/ |
D | mirror_pad_mode.cc | 24 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()
|
D | padding.cc | 24 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/ |
D | unpack_vectorizer.cc | 34 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/ |
D | encapsulate_util.cc | 47 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()
|
D | extract_outside_compilation_pass.cc | 80 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 …]
|
D | encapsulate_xla_computations_pass.cc | 47 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()
|
D | extract_outside_compilation_pass_test.cc | 79 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/ |
D | nccl_rewrite.cc | 33 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()
|