Lines Matching refs:element_type
120 PrimitiveType element_type, tensorflow::gtl::ArraySlice<int64> dimensions,
126 if (element_type == OPAQUE || element_type == TUPLE) {
128 PrimitiveType_Name(element_type).c_str());
130 Shape shape = ShapeUtil::MakeShape(element_type, dimensions);
183 PrimitiveType element_type, tensorflow::gtl::ArraySlice<int64> dimensions) {
184 DCHECK_NE(TUPLE, element_type);
185 DCHECK_NE(OPAQUE, element_type);
187 PopulateShape(element_type, dimensions, &result);
192 PrimitiveType element_type, tensorflow::gtl::ArraySlice<int64> dimensions,
194 return MakeShapeWithLayoutInternal(element_type, dimensions, minor_to_major)
199 PrimitiveType element_type, tensorflow::gtl::ArraySlice<int64> dimensions) {
202 return MakeShapeWithLayout(element_type, dimensions, layout);
206 PrimitiveType element_type, tensorflow::gtl::ArraySlice<int64> dimensions,
208 DCHECK_NE(TUPLE, element_type);
209 DCHECK_NE(OPAQUE, element_type);
210 Shape shape = ShapeUtil::MakeShape(element_type, dimensions);
223 return MakeShapeWithDescendingLayout(shape.element_type(), dims);
227 PrimitiveType element_type, tensorflow::gtl::ArraySlice<int64> dimensions,
230 shape->set_element_type(element_type);
270 return primitive_util::IsIntegralType(shape.element_type());
279 if (shape.element_type() == TUPLE || shape.element_type() == OPAQUE) {
282 return primitive_util::BitWidth(shape.element_type()) == bits;
286 switch (shape.element_type()) {
308 LOG(FATAL) << "Unhandled element type " << shape.element_type();
313 return primitive_util::IsComplexType(shape.element_type());
317 return primitive_util::IsFloatingPointType(shape.element_type());
363 complex_shape.element_type()));
367 PrimitiveType element_type,
369 return Equal(shape, MakeShape(element_type, dimensions));
385 return shape.element_type() == F32 && Rank(shape) == 0;
401 PrimitiveType_Name(shape.element_type())),
468 LowercasePrimitiveTypeName(shape.element_type()), "[");
616 if (lhs.element_type() == TUPLE) {
617 return rhs.element_type() == TUPLE &&
625 if (lhs.element_type() == TUPLE) {
626 return rhs.element_type() == TUPLE &&
635 if (lhs.element_type() == TUPLE) {
636 return rhs.element_type() == TUPLE &&
703 DCHECK_NE(OPAQUE, shape.element_type());
704 if (shape.element_type() == TUPLE) {
717 DCHECK_EQ(TUPLE, shape.element_type());
744 ByteSizeOfPrimitiveType(shape.element_type());
756 if (shape.element_type() == TUPLE) {
771 if (shape.element_type() == PRIMITIVE_TYPE_INVALID) {
886 shape.has_layout() ? MakeShapeWithLayout(shape.element_type(),
888 : MakeShape(shape.element_type(), dimension_sizes);
1236 input_shape.element_type(), AsInt64Slice(input_shape.dimensions()));
1238 output_shape.element_type(), AsInt64Slice(output_shape.dimensions()));
1397 output_shape.element_type(), AsInt64Slice(output_shape.dimensions()),
