Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.cc86 PrimitiveType primitive_type, absl::Span<const int64> dimensions) { in CreateFromDimensions()
125 /* static */ Literal LiteralUtil::Zero(PrimitiveType primitive_type) { in Zero()
166 /* static */ Literal LiteralUtil::One(PrimitiveType primitive_type) { in One()
207 /* static */ Literal LiteralUtil::MinValue(PrimitiveType primitive_type) { in MinValue()
252 /* static */ Literal LiteralUtil::MaxValue(PrimitiveType primitive_type) { in MaxValue()
294 PrimitiveType primitive_type) { in NanValue()
Dprimitive_util.cc180 bool IsArrayType(PrimitiveType primitive_type) { in IsArrayType()
Dshape_util.cc115 PrimitiveType primitive_type) { in IsArrayPrimitiveType()
587 PrimitiveType primitive_type) { in HasPrimitiveType()
738 PrimitiveType primitive_type) { in ByteSizeOfPrimitiveType()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dtype_to_shape.cc113 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape() local
125 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape() local
172 PrimitiveType primitive_type = TypeToPrimitiveType(element_type); in TypeToShape() local
/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.cc75 PrimitiveType primitive_type = input_shape.element_type(); in PotentiallyImplementedAsEigenConvolution() local
Ddot_op_emitter.cc347 PrimitiveType primitive_type = dot_info_.result_shape.element_type(); in EmitTiledLlvmIrGemm() local
386 PrimitiveType primitive_type = dot_info_.result_shape.element_type(); in EmitTiledLlvmIrGemv() local
Dvector_support_library.cc25 VectorSupportLibrary::VectorSupportLibrary(PrimitiveType primitive_type, in VectorSupportLibrary()
Dir_emitter.cc299 int IrEmitter::MinimumAlignmentForPrimitiveType(PrimitiveType primitive_type) { in MinimumAlignmentForPrimitiveType()
956 PrimitiveType primitive_type = lhs->shape().element_type(); in HandleConvolution() local
2516 PrimitiveType primitive_type = output_shape.element_type(); in EmitFastConcatenate() local
2618 PrimitiveType primitive_type, in EmitTransferElements()
3239 PrimitiveType primitive_type = operands[0]->shape().element_type(); in ElementTypesSameAndSupported() 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/
Dhlo_creation_utils_test.cc31 PrimitiveType primitive_type, absl::Span<const int64> input_shape_dims, in CreateModuleWithProgramShape()
46 PrimitiveType primitive_type, absl::Span<const int64> input_shape_dims, in CreateModuleWithProgramShape()
Drng_expander.cc44 PrimitiveType primitive_type = rng->shape().element_type(); in ConvertSmallFpRngToF32Rng() local
Dhlo_lexer.cc284 PrimitiveType primitive_type = in LexIdentifier() local
Dcollective_ops_utils.h252 PrimitiveType primitive_type; member
Dbatchnorm_expander.cc74 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
/external/skia/src/gpu/ops/
DDrawCustomMeshOp.cpp29 GrPrimitiveType primitive_type(SkCustomMesh::Mode mode) { in primitive_type() function

12