Home
last modified time | relevance | path

Searched refs:type_string (Results 1 – 25 of 161) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_context.cc80 const string type_string = DataTypeString(type); in GetOrCreateMax() local
81 VLOG(1) << "Building Max() for " << type_string; in GetOrCreateMax()
82 xla::XlaBuilder b("max<" + type_string + ">"); in GetOrCreateMax()
96 const string type_string = DataTypeString(type); in GetOrCreateMin() local
97 VLOG(1) << "Building Min() for " << type_string; in GetOrCreateMin()
98 xla::XlaBuilder b("min<" + type_string + ">"); in GetOrCreateMin()
112 const string type_string = DataTypeString(type); in GetOrCreateAdd() local
113 VLOG(1) << "Building Add() for " << type_string; in GetOrCreateAdd()
114 xla::XlaBuilder b("add<" + type_string + ">"); in GetOrCreateAdd()
128 const string type_string = DataTypeString(type); in GetOrCreateMul() local
[all …]
Dresource_util.cc75 const XlaResourceOpInfo* op_info = GetResourceOpInfoForOp(n.type_string()); in IsStackOrTensorArraySource()
89 n.type_string()); in PropagateFromStackOrTensorArraySourceOp()
104 TF_RET_CHECK(n.type_string() == kArgOp); in PropagateFromArgOp()
116 n.type_string()); in PropagateFromArgOp()
217 fbody->graph, n.type_string(), call_depth + 1, resource_arg_indices, in PropagateThroughCallOp()
231 TF_RET_CHECK(n.IsIdentity() || n.type_string() == kIdentityNOp); in PropagateThroughIdentityOp()
277 if (n->type_string() == kIfOp || n->type_string() == kWhileOp) { in AnalyzeResourceUsage()
307 if (n->IsIdentity() || n->type_string() == kIdentityNOp) { in AnalyzeResourceUsage()
314 it.first->dst()->type_string()); in AnalyzeResourceUsage()
Dfunctionalize_control_flow.cc106 if (n->type_string() == "Switch" || n->type_string() == "Merge") { in FunctionalizeControlFlowForFunction()
247 auto it = kNodeTypeToFunctionAttrMapping->find(n->type_string()); in Run()
254 VLOG(2) << "Graph has node " << n->type_string() in Run()
Dconst_analysis.cc187 if (XlaOpRegistry::IsMetadataOp(node->type_string())) { in BackwardsConstAnalysis()
194 if (node->type_string() == "_Arg") { in BackwardsConstAnalysis()
210 pred->src()->type_string() == "IdentityN") { in BackwardsConstAnalysis()
240 edge->src()->type_string() == "IdentityN") { in BackwardsConstAnalysis()
/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util.cc74 << node.type_string() << ")" << (reason.empty() ? "" : ": ") in LogNotCompilable()
138 if (node.type_string() == "SymbolicGradient") { in HasXLAKernel()
143 if (node.type_string() == "Const") { in HasXLAKernel()
296 return node.type_string() == "SelfAdjointEigV2" || in OpIsInaccurate()
297 node.type_string() == "Svd"; in OpIsInaccurate()
307 return node.type_string() == "SelfAdjointEigV2" || in OpIsSlow()
308 node.type_string() == "Svd" || node.type_string() == "Qr" || in OpIsSlow()
309 node.type_string() == "MatrixInverse" || in OpIsSlow()
310 node.type_string() == "MatrixSolve" || in OpIsSlow()
311 node.type_string() == "ResizeNearestNeighbor" || in OpIsSlow()
[all …]
Dxla_cluster_util.cc99 << incoming_node->name() << " " << incoming_node->type_string(); in HasForwardedRefInput()
306 if (flib_def->Contains(n.type_string())) { in MayCallFunction()
318 return node.type_string() == "Shape" || node.type_string() == "Rank" || in IsShapeConsumerOp()
319 node.type_string() == "Size"; in IsShapeConsumerOp()
366 info->op_histogram[n->type_string()]++; in GetXlaAutoClusteringSummary()
369 unclustered_op_histogram[n->type_string()]++; in GetXlaAutoClusteringSummary()
397 if (flib_def.Find(n.type_string())) { in GetCallTargetListFromNode()
399 callee.set_name(n.type_string()); in GetCallTargetListFromNode()
Dencapsulate_xla_computations_pass.cc77 return errors::InvalidArgument("Invalid ", n.type_string(), " number ", in GetIndexAttr()
128 if (n->type_string() == "_Arg") { in RewriteSubgraph()
135 } else if (n->type_string() == "_Retval") { in RewriteSubgraph()
221 e->dst()->type_string() != kXlaClusterOutput) { in Encapsulate()
308 TF_RET_CHECK(output_node->type_string() == kXlaClusterOutput) in BuildXlaLaunchOps()
337 function.set_name(launch->type_string()); in BuildXlaLaunchOps()
/external/mesa3d/src/mapi/glapi/gen/
DglX_proto_recv.py200 t = param.type_string()
267 type_string = param.type_string()
272 …rint '%s %s const %s = (%s) (%s(pc + %s));' % (indent, type_string, param.name, type_string, co…
282 print '%s const %s %s = %s;' % (indent, type_string, param.name, location)
287 print '%s %s %s = %s%s;' % (indent, type_string, param.name, cond, location)
289 print '%s %s %s;' % (indent, type_string, param.name)
342 … print ' %s = (%s) (pc + %s); break;' % (param.name, param.type_string(), o)
345 … %s = (%s) %s( (%s *) (pc + %s), %s ); break;' % (param.name, param.type_string(), swap_func, s…
352 …print ' %s = (%s) %s( (%s *) (pc + %s), %s );' % (param.name, param.type_string(), swap_func, s…
359 … print '%s %s = (%s) (pc + %s);' % (indent, param.name, param.type_string(), param.offset)
[all …]
Dtypeexpr.py116 def __init__(self, type_string, extra_types = None): argument
119 if not type_string:
122 self.original_string = type_string
129 tokens = type_string.replace("*", " * ").split()
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc1065 << n->type_string() << ", reason: " << reason; in CanOpRunOnCPUDevice()
1113 if (m->type_string() == csinfo_.bias_add) { in GetConv2DOrBiasAdd()
1117 CHECK_EQ(m->type_string(), csinfo_.conv2d); in GetConv2DOrBiasAdd()
1122 e->dst()->type_string() == csinfo_.bias_add && in GetConv2DOrBiasAdd()
1154 if (m->type_string() == csinfo_.pad) { in GetPadOrConv2D()
1157 if (!e->IsControlEdge() && e->dst()->type_string() == csinfo_.conv2d) { in GetPadOrConv2D()
1164 DCHECK_EQ(m->type_string(), csinfo_.conv2d); in GetPadOrConv2D()
1168 if (!e->IsControlEdge() && e->src()->type_string() == csinfo_.pad) { in GetPadOrConv2D()
1203 if (m->type_string() == csinfo_.pad) { in GetPadOrFusedConv2D()
1207 e->dst()->type_string() == csinfo_.fused_conv2d) { in GetPadOrFusedConv2D()
[all …]
Dmkl_tfconversion_pass.cc201 << src->type_string() << " and " << dst->type_string() in InsertConversionNodeOnEdge()
289 << "conversion node on: " << n->type_string() << " successful."; in InsertInputConversionNode()
325 if (src->type_string().compare("_MklToTf") == 0 || in RunPass()
326 dst->type_string().compare("_MklToTf") == 0) { in RunPass()
331 << src->type_string() << " and " << dst->type_string(); in RunPass()
340 IsMklSupportedOp(src->type_string(), src_datatype)); in RunPass()
343 IsMklSupportedOp(dst->type_string(), dst_datatype)); in RunPass()
383 if (IsMklElementWiseOp(n->type_string(), datatype)) { in RunPass()
389 (input_node->type_string().compare("_MklInputConversion") == 0)) { in RunPass()
Doptimizer_cse.cc201 hasher.MixString(n->type_string()); in NodeHash()
242 if (a->type_string() != b->type_string()) return false; in Equivalent()
300 if (n->type_string() == "Placeholder" || in Optimize()
301 n->type_string() == "PlaceholderV2" || in Optimize()
302 n->type_string() == "PlaceholderWithDefault") { in Optimize()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dstrip_unused_nodes.cc43 const string& type_string = context.params.at("type")[0]; in TypeForPlaceholder() local
44 if (!DataTypeFromString(type_string, result)) { in TypeForPlaceholder()
46 type_string, "'"); in TypeForPlaceholder()
64 const string& type_string = context.params.at("type_for_name")[i]; in TypeForPlaceholder() local
65 if (!DataTypeFromString(type_string, result)) { in TypeForPlaceholder()
67 type_string, "'"); in TypeForPlaceholder()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_while_op_test.cc135 if (op->type_string() == "LessThanOrEqualToN") { in TEST()
138 if (op->type_string() == "XTimesTwo") { in TEST()
144 ASSERT_NE(op->type_string(), "While"); in TEST()
240 if (op->type_string() == "XPlusOneXTimesY") { in TEST()
243 if (op->type_string() == "XYXLessThanOrEqualToN") { in TEST()
246 ASSERT_NE(op->type_string(), "While"); in TEST()
326 if (op->type_string() == x_times_two.signature().name()) { in TEST()
329 if (op->type_string() == less_than_or_eq.signature().name()) { in TEST()
332 ASSERT_NE(op->type_string(), "While"); in TEST()
Deval_const_tensor.cc51 if (node->type_string() == "Shape") { in TryToInferTensorOutputFromInputShapes()
81 } else if (node->type_string() == "Rank") { in TryToInferTensorOutputFromInputShapes()
90 } else if (node->type_string() == "Size") { in TryToInferTensorOutputFromInputShapes()
145 if (target_node.type_string() == "PlaceholderWithDefault") { in ExtractConstantSubgraph()
215 if (current_node->type_string() == "PlaceholderWithDefault") { in ExtractConstantSubgraph()
Dlower_if_op_test.cc118 ASSERT_NE(op->type_string(), "If"); in TEST()
228 ASSERT_NE(op->type_string(), "If"); in TEST()
307 if (op->type_string() == x_times_two.signature().name()) { in TEST()
310 if (op->type_string() == x_times_four.signature().name()) { in TEST()
313 ASSERT_NE(op->type_string(), "If"); in TEST()
/external/tensorflow/tensorflow/core/framework/
Dop_def_builder.cc128 bool ProcessCompoundType(const StringPiece type_string, AttrValue* allowed) { in ProcessCompoundType() argument
129 if (type_string == "numbertype" || type_string == "numerictype") { in ProcessCompoundType()
133 } else if (type_string == "quantizedtype") { in ProcessCompoundType()
137 } else if (type_string == "realnumbertype" || in ProcessCompoundType()
138 type_string == "realnumerictype") { in ProcessCompoundType()
161 StringPiece type_string; // Used if type == "type" in FinalizeAttr() local
178 } else if (ConsumeCompoundAttrType(&spec, &type_string)) { in FinalizeAttr()
181 VERIFY(ProcessCompoundType(type_string, allowed), in FinalizeAttr()
182 "Expected to see a compound type, saw: ", type_string); in FinalizeAttr()
213 VERIFY(ConsumeAttrType(&spec, &type_string), in FinalizeAttr()
[all …]
Ddataset.cc215 const string& type_string = dataset->type_string(); in AddDataset() local
220 bool has_output_types_attr = HasAttr(type_string, "output_types"); in AddDataset()
221 bool has_output_shapes_attr = HasAttr(type_string, "output_shapes"); in AddDataset()
238 NodeBuilder node_builder(opts->GetNameForOp(type_string), type_string, in AddDataset()
264 return errors::Internal("AddDataset: Failed to build ", type_string, in AddDataset()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dconv_op_helpers.h61 StringPiece type_string, xla::XlaOp conv_input, xla::XlaOp filter,
65 StringPiece type_string, const xla::Shape& input_shape, xla::XlaOp filter,
69 StringPiece type_string, xla::XlaOp activations,
Dconv_ops.cc56 ctx->op_kernel().type_string(), ctx->Input(0), ctx->Input(1), attrs_); in Compile()
112 MakeXlaBackpropInputConvOp(ctx->op_kernel().type_string(), input_shape, in Compile()
173 ctx->op_kernel().type_string(), ctx->Input(0), filter_shape, in Compile()
/external/adhd/cras/src/server/
Dcras_dbus_util.c9 const char *type_string, void *value) in append_key_value() argument
19 type_string, &variant)) in append_key_value()
/external/tensorflow/tensorflow/core/kernels/
Ddecode_image_op.cc77 if (type_string() == "DecodeJpeg") { in DecodeImageOp()
79 } else if (type_string() == "DecodeAndCropJpeg") { in DecodeImageOp()
82 } else if (type_string() == "DecodePng") { in DecodeImageOp()
84 } else if (type_string() == "DecodeGif") { in DecodeImageOp()
88 errors::InvalidArgument("Bad op type ", type_string())); in DecodeImageOp()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc444 } else if (ops_definitions.GetOpIdFor(node.type_string(), {}) != in RegisterNode()
449 return errors::InvalidArgument(node.type_string() + in RegisterNode()
589 const StringPiece op_type(node.type_string()); in NeedsToAddRank()
597 const StringPiece op_type(node.type_string()); in IsPadNode()
607 if (node.type_string() != RESHAPE_NODE_TYPE_STRING) { in IsNodeFlattenReshape()
665 const int op_type_id = ops_definitions.GetOpIdFor(node.type_string(), {}); in RegisterNodeWithPaddingAndStrides()
667 << "Op " << node.type_string() << " not found in map(id = " << op_type_id in RegisterNodeWithPaddingAndStrides()
672 shape_refiner, node, node.name(), id, node.type_string(), op_type_id, in RegisterNodeWithPaddingAndStrides()
694 const int op_type_id = ops_definitions.GetOpIdFor(node.type_string(), {}); in RegisterNodeWithRank()
696 << "Op " << node.type_string() << " not found in map(id = " << op_type_id in RegisterNodeWithRank()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/vectorization/
Ddecode_csv_vectorizer.cc40 auto node_builder = NodeBuilder(node.type_string(), node.type_string()) in Vectorize()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
Dsegment.cc430 << "(Op type: " << node->tf_node()->type_string() << "), " in SegmentGraph()
433 unsupported_ops.emplace(node->tf_node()->type_string()); in SegmentGraph()
440 << "(Op type: " << node->tf_node()->type_string() << "), " in SegmentGraph()
443 unsupported_ops.emplace(node->tf_node()->type_string()); in SegmentGraph()
448 << "(Op type: " << node->tf_node()->type_string() << "), " in SegmentGraph()
644 for (auto in : (is_input_nodes || node->type_string() == "Const") in SegmentGraph()
678 StrAppend(&s, "\n[Op type: ", node->type_string(), "] ", node->name()); in SegmentGraph()
688 return noops->count(node->type_string()) == 0; in SegmentGraph()

1234567