Home
last modified time | relevance | path

Searched refs:primitive_type (Results 1 – 25 of 56) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/xla/python_api/
Dtypes.py46 primitive_type=xla_data_pb2.BF16,
52 primitive_type=xla_data_pb2.F16,
58 primitive_type=xla_data_pb2.F32,
64 primitive_type=xla_data_pb2.F64,
70 primitive_type=xla_data_pb2.S8,
76 primitive_type=xla_data_pb2.S16,
82 primitive_type=xla_data_pb2.S32,
88 primitive_type=xla_data_pb2.S64,
94 primitive_type=xla_data_pb2.U8,
100 primitive_type=xla_data_pb2.U16,
[all …]
Dxla_shape.py112 element_type = types.MAP_DTYPE_TO_RECORD[str(ndarray.dtype)].primitive_type
157 element_type = types.MAP_DTYPE_TO_RECORD[str(dtype)].primitive_type
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dtype_to_shape.cc114 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape() local
115 if (primitive_type != PrimitiveType::PRIMITIVE_TYPE_INVALID) in TypeToShape()
116 return ShapeUtil::MakeShape(primitive_type, span); in TypeToShape()
130 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape() local
131 if (primitive_type == PrimitiveType::PRIMITIVE_TYPE_INVALID) break; in TypeToShape()
139 return ShapeUtil::MakeShape(primitive_type, span); in TypeToShape()
154 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape() local
156 if (primitive_type != PrimitiveType::PRIMITIVE_TYPE_INVALID) in TypeToShape()
157 return ShapeUtil::MakeShape(primitive_type, span); in TypeToShape()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc86 PrimitiveType primitive_type, absl::Span<const int64> dimensions) { in CreateFromDimensions() argument
88 ShapeUtil::MakeShape(primitive_type, dimensions)); in CreateFromDimensions()
125 /* static */ Literal LiteralUtil::Zero(PrimitiveType primitive_type) { in Zero() argument
126 switch (primitive_type) { in Zero()
162 LOG(FATAL) << "Unhandled primitive type " << primitive_type; in Zero()
166 /* static */ Literal LiteralUtil::One(PrimitiveType primitive_type) { in One() argument
167 switch (primitive_type) { in One()
203 LOG(FATAL) << "Unhandled primitive type " << primitive_type; in One()
207 /* static */ Literal LiteralUtil::MinValue(PrimitiveType primitive_type) { in MinValue() argument
208 switch (primitive_type) { in MinValue()
[all …]
Dprimitive_util.cc127 bool IsArrayType(PrimitiveType primitive_type) { in IsArrayType() argument
128 return primitive_type != PRIMITIVE_TYPE_INVALID && primitive_type != TUPLE && in IsArrayType()
129 primitive_type != OPAQUE_TYPE && primitive_type != TOKEN; in IsArrayType()
Dliteral_util.h105 static Literal Zero(PrimitiveType primitive_type);
107 static Literal One(PrimitiveType primitive_type);
110 static Literal MinValue(PrimitiveType primitive_type);
113 static Literal MaxValue(PrimitiveType primitive_type);
117 static StatusOr<Literal> NanValue(PrimitiveType primitive_type);
213 static Literal CreateFromDimensions(PrimitiveType primitive_type,
Dshape_util.cc84 PrimitiveType primitive_type) { in IsArrayPrimitiveType() argument
85 return primitive_util::IsArrayType(primitive_type); in IsArrayPrimitiveType()
449 PrimitiveType primitive_type) { in HasPrimitiveType() argument
450 if (shape.element_type() == primitive_type) { in HasPrimitiveType()
454 if (HasPrimitiveType(element_shape, primitive_type)) { in HasPrimitiveType()
579 PrimitiveType primitive_type) { in ByteSizeOfPrimitiveType() argument
580 switch (primitive_type) { in ByteSizeOfPrimitiveType()
616 LOG(FATAL) << PrimitiveType_Name(primitive_type) in ByteSizeOfPrimitiveType()
619 LOG(FATAL) << "Unhandled primitive type " << primitive_type; in ByteSizeOfPrimitiveType()
Dprimitive_util.h149 bool IsArrayType(PrimitiveType primitive_type);
/external/mesa3d/src/gallium/drivers/etnaviv/
Detnaviv_emit.h96 etna_draw_primitives(struct etna_cmd_stream *stream, uint32_t primitive_type, in etna_draw_primitives() argument
102 etna_cmd_stream_emit(stream, primitive_type); in etna_draw_primitives()
109 uint32_t primitive_type, uint32_t start, in etna_draw_indexed_primitives() argument
115 etna_cmd_stream_emit(stream, primitive_type); in etna_draw_indexed_primitives()
125 uint32_t indexed, uint32_t primitive_type, in etna_draw_instanced() argument
133 VIV_FE_DRAW_INSTANCED_HEADER_TYPE(primitive_type) | in etna_draw_instanced()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_eigen_dot_operation_test.cc33 PrimitiveType primitive_type; member
38 return PrimitiveType_Name(info.param.primitive_type); in DotTestSpecToString()
65 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128}); in TEST_P()
80 auto param_shape = ShapeUtil::MakeShape(spec.primitive_type, {128, 128}); in TEST_P()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dtpu_driver.cc36 int64_t ByteSizeOfPrimitiveType(xla::PrimitiveType primitive_type) { in ByteSizeOfPrimitiveType() argument
37 switch (primitive_type) { in ByteSizeOfPrimitiveType()
71 LOG(FATAL) << PrimitiveType_Name(primitive_type) in ByteSizeOfPrimitiveType()
74 LOG(FATAL) << "Unhandled primitive type " << primitive_type; in ByteSizeOfPrimitiveType()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc75 PrimitiveType primitive_type = input_shape.element_type(); in PotentiallyImplementedAsEigenConvolution() local
76 if (primitive_type != F16 && primitive_type != F32) { in PotentiallyImplementedAsEigenConvolution()
Ddot_op_emitter.cc223 PrimitiveType primitive_type = dot_info_.result_shape.element_type(); in EmitTiledLlvmIrGemm() local
238 int64 size_bytes = m * n * ShapeUtil::ByteSizeOfPrimitiveType(primitive_type); in EmitTiledLlvmIrGemm()
244 *b_->GetInsertBlock()->getParent(), primitive_type); in EmitTiledLlvmIrGemm()
251 /*scalar_type=*/primitive_type, in EmitTiledLlvmIrGemm()
261 PrimitiveType primitive_type = dot_info_.result_shape.element_type(); in EmitTiledLlvmIrGemv() local
263 CHECK(primitive_util::IsFloatingPointType(primitive_type) || in EmitTiledLlvmIrGemv()
264 primitive_util::IsIntegralType(primitive_type)); in EmitTiledLlvmIrGemv()
341 *b_->GetInsertBlock()->getParent(), primitive_type); in EmitTiledLlvmIrGemv()
356 /*scalar_type=*/primitive_type, in EmitTiledLlvmIrGemv()
365 /*scalar_type=*/primitive_type, in EmitTiledLlvmIrGemv()
/external/tensorflow/tensorflow/compiler/xla/service/
Drng_expander.cc44 PrimitiveType primitive_type = rng->shape().element_type(); in ConvertSmallFpRngToF32Rng() local
45 CHECK(primitive_type == F16 || primitive_type == BF16); in ConvertSmallFpRngToF32Rng()
50 CHECK_EQ(operand->shape().element_type(), primitive_type); in ConvertSmallFpRngToF32Rng()
62 rng->ReplaceAllUsesWith(MakeConvertToHlo(new_rng, primitive_type))); in ConvertSmallFpRngToF32Rng()
Dhlo_creation_utils_test.cc31 PrimitiveType primitive_type, absl::Span<const int64> input_shape_dims, in CreateModuleWithProgramShape() argument
34 Shape input_shape = ShapeUtil::MakeShape(primitive_type, input_shape_dims); in CreateModuleWithProgramShape()
36 ShapeUtil::MakeShape(primitive_type, output_shape_dims); in CreateModuleWithProgramShape()
46 PrimitiveType primitive_type, absl::Span<const int64> input_shape_dims, in CreateModuleWithProgramShape() argument
49 Shape input_shape = ShapeUtil::MakeShape(primitive_type, input_shape_dims); in CreateModuleWithProgramShape()
Dhlo_lexer.cc258 PrimitiveType primitive_type = in LexIdentifier() local
260 if (primitive_type != TUPLE) { in LexIdentifier()
261 token_state_.primitive_type_val = primitive_type; in LexIdentifier()
/external/deqp/external/openglcts/modules/gl/
Dgl3cClipDistance.cpp584 glw::GLenum primitive_type = m_primitive_types[i_primitive_type]; in iterate() local
588 glw::GLuint framebuffer_size = (primitive_type == GL_POINTS) ? 1 : 32; in iterate()
613 redeclaration, redeclaration, i_clip_count, i_clip_function, primitive_type); in iterate()
641 gl3cts::ClipDistance::Utility::VertexArrayObject vertex_array_object(gl, primitive_type); in iterate()
644 prepareGeometry(gl, primitive_type); in iterate()
665 if (!checkResults(primitive_type, i_clip_function, results)) in iterate()
669 << glu::getPrimitiveTypeStr(primitive_type) in iterate()
714 glw::GLenum primitive_type) in prepareVertexShaderCode() argument
757 switch (primitive_type) in prepareVertexShaderCode()
781 const glw::Functions& gl, const glw::GLenum primitive_type) in prepareGeometry() argument
[all …]
Dgl3cTransformFeedbackTests.hpp424 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
504 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
538 virtual void draw(glw::GLenum primitive_type) in draw() argument
540 CaptureGeometryInterleaved::draw(primitive_type); in draw()
558 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type) in checkTransformFeedbackBuffer() argument
560 return CaptureVertexSeparate::checkTransformFeedbackBuffer(bind_case, primitive_type); in checkTransformFeedbackBuffer()
705 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
737 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
771 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
806 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type);
Dgl3cClipDistance.hpp144 VertexArrayObject(const glw::Functions& gl, const glw::GLenum primitive_type); // create empty vao
263 glw::GLuint clip_function, glw::GLenum primitive_type);
266 const glw::GLenum primitive_type);
268 …bool checkResults(glw::GLenum primitive_type, glw::GLuint clip_function, std::vector<glw::GLfloat>…
Dgl4cPipelineStatisticsQueryTests.cpp603 …w::GLenum PipelineStatisticsQueryUtilities::getEnumForPrimitiveType(_primitive_type primitive_type) in getEnumForPrimitiveType() argument
607 switch (primitive_type) in getEnumForPrimitiveType()
797 …pelineStatisticsQueryUtilities::getNumberOfVerticesForPrimitiveType(_primitive_type primitive_type) in getNumberOfVerticesForPrimitiveType() argument
801 switch (primitive_type) in getNumberOfVerticesForPrimitiveType()
985 …:string PipelineStatisticsQueryUtilities::getStringForPrimitiveType(_primitive_type primitive_type) in getStringForPrimitiveType() argument
989 switch (primitive_type) in getStringForPrimitiveType()
2236 glw::GLenum primitive_type = in queryCallbackDrawCallHandler() local
2243 gl.drawArrays(primitive_type, pInstance->m_indirect_draw_call_first_argument, in queryCallbackDrawCallHandler()
2253 gl.drawArraysIndirect(primitive_type, in queryCallbackDrawCallHandler()
2263 gl.drawArraysInstanced(primitive_type, pInstance->m_indirect_draw_call_first_argument, in queryCallbackDrawCallHandler()
[all …]
Dgl4cPipelineStatisticsQueryTests.hpp267 static glw::GLenum getEnumForPrimitiveType(_primitive_type primitive_type);
272 static unsigned int getNumberOfVerticesForPrimitiveType(_primitive_type primitive_type);
276 static std::string getStringForPrimitiveType(_primitive_type primitive_type);
/external/tensorflow/tensorflow/compiler/tf2xla/
Dliteral_util.cc83 xla::PrimitiveType primitive_type; in CopyLiteralToHostTensor() local
85 DataTypeToPrimitiveType(host_tensor->dtype(), &primitive_type)); in CopyLiteralToHostTensor()
86 if (literal.shape().element_type() != primitive_type) { in CopyLiteralToHostTensor()
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Drandom.cc71 xla::PrimitiveType primitive_type = in ParameterizedTruncatedNormal() local
73 xla::XlaOp epsilon = xla::Epsilon(uniform.builder(), primitive_type); in ParameterizedTruncatedNormal()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dprng.cc520 PrimitiveType primitive_type = shape.element_type(); in NormalFloatingPointDistribution() local
521 DCHECK(primitive_type == F32 || primitive_type == F64); in NormalFloatingPointDistribution()
528 xla::ConstantR0WithType(builder, primitive_type, 0.0), in NormalFloatingPointDistribution()
529 xla::ConstantR0WithType(builder, primitive_type, 1.0), in NormalFloatingPointDistribution()
530 ShapeUtil::MakeShape(primitive_type, {num_pairs * 2})); in NormalFloatingPointDistribution()
/external/ImageMagick/coders/
Dsvg.c4124 primitive_type; in WriteSVGImage() local
4221 primitive_type=UndefinedPrimitive; in WriteSVGImage()
4257 primitive_type=AlphaPrimitive; in WriteSVGImage()
4269 primitive_type=ArcPrimitive; in WriteSVGImage()
4280 primitive_type=BezierPrimitive; in WriteSVGImage()
4315 primitive_type=CirclePrimitive; in WriteSVGImage()
4320 primitive_type=ColorPrimitive; in WriteSVGImage()
4345 primitive_type=EllipsePrimitive; in WriteSVGImage()
4454 primitive_type=ImagePrimitive; in WriteSVGImage()
4485 primitive_type=LinePrimitive; in WriteSVGImage()
[all …]

123