Searched refs:shape_name (Results 1 – 3 of 3) sorted by relevance
205 string shape_name = strings::StrCat(in_node.node->name(), "/Shape"); in ReplaceBroadcast() local208 NodeBuilder shape_builder(shape_name, "Shape"); in ReplaceBroadcast()222 NodeBuilder shape_builder(strings::StrCat(shape_name, recv_index), in ReplaceBroadcast()
508 const string shape_name = CONST_SHAPE_PREFIX + ToString(shape.at(0)) + 'x' + in RegisterConstantShape() local511 if (node_name_to_id_cache_map_.count(shape_name) <= 0) { in RegisterConstantShape()514 node_name_to_id_cache_map_.emplace(shape_name, id); in RegisterConstantShape()517 const_node_info.set_name(shape_name); in RegisterConstantShape()525 return node_name_to_id_cache_map_[shape_name]; in RegisterConstantShape()
2178 shape_name = (name + "/shape") if name is not None else None2179 shape, rank = _normalize_sparse_shape(shape, shape_name)2181 shape = placeholder(dtypes.int64, shape=[rank], name=shape_name)