Searched refs:LowercasePrimitiveTypeName (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | operand_upcaster_test.cc | 51 module_tmpl, primitive_util::LowercasePrimitiveTypeName(lhs_type), in TEST_P()
52 primitive_util::LowercasePrimitiveTypeName(rhs_type), in TEST_P()
53 primitive_util::LowercasePrimitiveTypeName(result_type)); in TEST_P()
66 primitive_util::LowercasePrimitiveTypeName(result_type)))) in TEST_P()
73 primitive_util::LowercasePrimitiveTypeName(result_type)))) in TEST_P()
80 primitive_util::LowercasePrimitiveTypeName(result_type))))); in TEST_P()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | primitive_util.cc | 209 const string& LowercasePrimitiveTypeName(PrimitiveType s) { in LowercasePrimitiveTypeName() function
226 (*map)[LowercasePrimitiveTypeName(value)] = value; in GetPrimitiveTypeStringMap()
|
| D | primitive_util.h | 248 const string& LowercasePrimitiveTypeName(PrimitiveType s);
|
| D | shape_util.cc | 600 primitive_util::LowercasePrimitiveTypeName(shape.element_type()), "[", in HumanString()
806 primitive_util::LowercasePrimitiveTypeName(shape.element_type()), in ValidateShapeWithOptionalLayoutInternal()
812 primitive_util::LowercasePrimitiveTypeName(shape.element_type()), in ValidateShapeWithOptionalLayoutInternal()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_runner.cc | 249 primitive_util::LowercasePrimitiveTypeName(config.output_type) in RunCudnnBatchNormForwardInference()
278 primitive_util::LowercasePrimitiveTypeName(config.output_type) in RunCudnnBatchNormForwardTraining()
310 primitive_util::LowercasePrimitiveTypeName(config.output_type) in RunCudnnBatchNormBackward()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/xla/ |
| D | host_compute_ops.cc | 159 xla::primitive_util::LowercasePrimitiveTypeName( in Compile()
219 xla::primitive_util::LowercasePrimitiveTypeName( in Compile()
424 xla::primitive_util::LowercasePrimitiveTypeName( in Compile()
476 xla::primitive_util::LowercasePrimitiveTypeName( in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | stateless_random_ops.cc | 108 xla::primitive_util::LowercasePrimitiveTypeName(type))); in StatelessRngUniform()
137 xla::primitive_util::LowercasePrimitiveTypeName(type))); in StatelessRngUniformFullInt()
|
| D | stateful_random_ops.cc | 87 xla::primitive_util::LowercasePrimitiveTypeName(type))), in StatefulRngUniform()
110 xla::primitive_util::LowercasePrimitiveTypeName(type))), in StatefulRngUniformFullInt()
|
| D | stateless_random_ops_v2.cc | 138 xla::primitive_util::LowercasePrimitiveTypeName(type))), in StatelessRngUniformV2()
165 xla::primitive_util::LowercasePrimitiveTypeName(type))), in StatelessRngUniformFullInt()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 548 primitive_util::LowercasePrimitiveTypeName(type))), in ThreeFryBitGenerator()
569 primitive_util::LowercasePrimitiveTypeName(type))), in PhiloxBitGenerator()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | collective_ops_test.cc | 91 std::string dtype = primitive_util::LowercasePrimitiveTypeName( in TestTwoReplicasOneOperand()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_communication.cc | 240 ::xla::primitive_util::LowercasePrimitiveTypeName(xla_element_type); in SetFrontendAttributes()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 1127 primitive_util::LowercasePrimitiveTypeName(datatype)); in HandleAllReduceMultipleReplica()
|