Home
last modified time | relevance | path

Searched defs:primitive_type (Results 1 – 25 of 40) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dtype_to_shape.cc97 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape() local
110 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape() local
171 PrimitiveType primitive_type = TypeToPrimitiveType(t.getElementType()); in TypeToShape() local
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc92 Literal CreateScalar(PrimitiveType primitive_type, Args... args) { in CreateScalar()
275 PrimitiveType primitive_type, absl::Span<const int64_t> dimensions) { in CreateFromDimensions()
319 /* static */ Literal LiteralUtil::Zero(PrimitiveType primitive_type) { in Zero()
323 /* static */ Literal LiteralUtil::One(PrimitiveType primitive_type) { in One()
327 /* static */ Literal LiteralUtil::MinValue(PrimitiveType primitive_type) { in MinValue()
331 /* static */ Literal LiteralUtil::MaxValue(PrimitiveType primitive_type) { in MaxValue()
336 PrimitiveType primitive_type) { in NanValue()
Dtfrt_utils_test.cc28 PrimitiveType primitive_type; member
Dprimitive_util.h151 inline constexpr bool IsArrayType(PrimitiveType primitive_type) { in IsArrayType()
Dshape_util.cc92 PrimitiveType primitive_type) { in IsArrayPrimitiveType()
621 PrimitiveType primitive_type) { in HasPrimitiveType()
779 PrimitiveType primitive_type) { in ByteSizeOfPrimitiveType()
/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()
109 uint32_t primitive_type, uint32_t start, in etna_draw_indexed_primitives()
125 uint32_t indexed, uint32_t primitive_type, in etna_draw_instanced()
/external/deqp/external/openglcts/modules/gl/
Dgl3cClipDistance.cpp584 glw::GLenum primitive_type = m_primitive_types[i_primitive_type]; in iterate() local
714 glw::GLenum primitive_type) in prepareVertexShaderCode()
781 const glw::Functions& gl, const glw::GLenum primitive_type) in prepareGeometry()
832 bool gl3cts::ClipDistance::FunctionalTest::checkResults(glw::GLenum primitive_type, glw::GLuint cli… in checkResults()
1578 const glw::GLenum primitive_type) in VertexArrayObject()
Dgl4cPipelineStatisticsQueryTests.cpp617 …w::GLenum PipelineStatisticsQueryUtilities::getEnumForPrimitiveType(_primitive_type primitive_type) in getEnumForPrimitiveType()
811 …pelineStatisticsQueryUtilities::getNumberOfVerticesForPrimitiveType(_primitive_type primitive_type) in getNumberOfVerticesForPrimitiveType()
999 …:string PipelineStatisticsQueryUtilities::getStringForPrimitiveType(_primitive_type primitive_type) in getStringForPrimitiveType()
2266 glw::GLenum primitive_type = in queryCallbackDrawCallHandler() local
Dgl3cTransformFeedbackTests.cpp2579 glw::GLenum primitive_type) in checkTransformFeedbackBuffer()
3220 glw::GLenum primitive_type) in checkTransformFeedbackBuffer()
3803 glw::GLenum primitive_type) in checkTransformFeedbackBuffer()
3877 glw::GLenum primitive_type) in checkTransformFeedbackBuffer()
3950 glw::GLenum primitive_type) in checkTransformFeedbackBuffer()
4024 glw::GLenum primitive_type) in checkTransformFeedbackBuffer()
Dgl3cTransformFeedbackTests.hpp538 virtual void draw(glw::GLenum primitive_type) in draw()
558 virtual bool checkTransformFeedbackBuffer(BindBufferCase bind_case, glw::GLenum primitive_type) in checkTransformFeedbackBuffer()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dtpu_driver.cc36 int64_t ByteSizeOfPrimitiveType(xla::PrimitiveType primitive_type) { in ByteSizeOfPrimitiveType()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_emission_utils.cc74 PrimitiveType primitive_type = input_shape.element_type(); in PotentiallyImplementedAsEigenConvolution() local
Dvector_support_library.cc25 VectorSupportLibrary::VectorSupportLibrary(PrimitiveType primitive_type, in VectorSupportLibrary()
Dir_emitter.cc326 int IrEmitter::MinimumAlignmentForPrimitiveType(PrimitiveType primitive_type) { in MinimumAlignmentForPrimitiveType()
990 PrimitiveType primitive_type = lhs->shape().element_type(); in HandleConvolution() local
2654 PrimitiveType primitive_type = output_shape.element_type(); in EmitFastConcatenate() local
2757 PrimitiveType primitive_type, in EmitTransferElements()
3406 PrimitiveType primitive_type = operands[0]->shape().element_type(); in ElementTypesSameAndSupported() local
Ddot_op_emitter.cc380 PrimitiveType primitive_type = dot_info_.result_shape.element_type(); in EmitTiledLlvmIrGemm() local
419 PrimitiveType primitive_type = dot_info_.result_shape.element_type(); in EmitTiledLlvmIrGemv() local
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Drandom.cc71 xla::PrimitiveType primitive_type = in ParameterizedTruncatedNormal() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_eigen_dot_operation_test.cc34 PrimitiveType primitive_type; member
/external/tensorflow/tensorflow/compiler/tf2xla/
Dliteral_util.cc94 xla::PrimitiveType primitive_type; in CopyLiteralToHostTensor() local
/external/tensorflow/tensorflow/compiler/xla/service/
Drng_expander.cc46 PrimitiveType primitive_type = rng->shape().element_type(); in ConvertSmallFpRngToF32Rng() local
Dhlo_creation_utils_test.cc33 PrimitiveType primitive_type, absl::Span<const int64_t> input_shape_dims, in CreateModuleWithProgramShape()
48 PrimitiveType primitive_type, absl::Span<const int64_t> input_shape_dims, in CreateModuleWithProgramShape()
Dhlo_lexer.cc283 PrimitiveType primitive_type = in LexIdentifier() local
Dcollective_ops_utils.h260 PrimitiveType primitive_type; member
Dbatchnorm_expander.cc73 PrimitiveType primitive_type) { in GetOrCreateScalarAddComputation()
/external/rust/crates/jni/src/wrapper/objects/
Djvalue.rs74 pub fn primitive_type(&self) -> Option<Primitive> { in primitive_type() method
/external/tensorflow/tensorflow/java/src/main/native/
Dtensor_jni.cc164 #define CASE(dtype, jtype, primitive_type) \ in read1DArray() argument

12