| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 39 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in Zero()
45 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in One()
52 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in IntegerLiteral()
59 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in FloatLiteral()
120 TF_CHECK_OK(DataTypeToPrimitiveType(new_element_type, &convert_to)); in ConvertElementType()
|
| D | xla_context.cc | 84 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateMax()
100 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateMin()
116 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateAdd()
132 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateMul()
|
| D | type_util.h | 26 Status DataTypeToPrimitiveType(DataType data_type, xla::PrimitiveType* type);
|
| D | type_util.cc | 23 Status DataTypeToPrimitiveType(DataType data_type, xla::PrimitiveType* type) { in DataTypeToPrimitiveType() function
|
| D | literal_util.cc | 85 DataTypeToPrimitiveType(host_tensor->dtype(), &primitive_type)); in CopyLiteralToHostTensor()
|
| D | shape_util.cc | 105 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(dtype, &type)); in TensorShapeToXLAShape()
|
| D | xla_op_kernel.cc | 105 Status status = DataTypeToPrimitiveType(input_type(index), &type); in input_xla_type()
473 Status status = DataTypeToPrimitiveType(expected_output_dtype(index), &type); in output_xla_type()
|
| D | xla_compiler_test.cc | 1209 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(type, &ptype)); in TEST_F()
1279 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(type, &ptype)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cast_op.cc | 34 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(src_dtype_, &src_type_)); in CastOp()
35 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dst_dtype_, &dst_type_)); in CastOp()
105 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(src_dtype_, &src_type_)); in BitcastOp()
106 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dst_dtype_, &dst_type_)); in BitcastOp()
|
| D | categorical_op.cc | 71 DataTypeToPrimitiveType(input_type(0), &uniform_xla_type)); in Compile()
83 DataTypeToPrimitiveType(input_type(0), &uniform_xla_type)); in Compile()
89 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_type(0), &type)); in Compile()
102 DataTypeToPrimitiveType(output_type(0), &xla_output_type)); in Compile()
|
| D | reduction_ops_common.cc | 35 ctx, DataTypeToPrimitiveType(reduction_type_, &xla_reduction_type_)); in XlaReductionOp()
107 TF_CHECK_OK(DataTypeToPrimitiveType(reduction_type_, &type)); in Compile()
|
| D | pooling_ops.cc | 69 ctx, DataTypeToPrimitiveType(reduction_type_, &xla_reduction_type_)); in PoolingOp()
333 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_type(2), &element_type)); in Compile()
426 ctx, DataTypeToPrimitiveType(reduction_type, &xla_reduction_type)); in Compile()
440 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(ctx->input_type(1), in Compile()
613 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_type(2), &element_type)); in Compile()
|
| D | empty_op.cc | 35 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype_, &type_)); in EmptyOp()
|
| D | batch_norm_op.cc | 45 DataTypeToPrimitiveType(ctx->input_type(0), &input_type)); in Compile()
48 DataTypeToPrimitiveType(ctx->input_type(1), &scale_type)); in Compile()
|
| D | index_ops.cc | 63 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(index_type, &index_xla_type)); in Compile()
|
| D | index_ops_cpu.cc | 71 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype, &output_type)); in Compile()
|
| D | segment_reduction_ops.cc | 32 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype, &type_)); in UnsortedSegmentReduce()
|
| D | quantize_and_dequantize_op.cc | 57 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(data_type, &xla_type)); in Compile()
|
| D | extract_image_patches_op.cc | 88 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(ctx->input_type(0), &type)); in Compile()
|
| D | softmax_op.cc | 67 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(accumulation_type, in Compile()
|
| D | stateful_random_ops.cc | 236 DataTypeToPrimitiveType(STATE_ELEMENT_DTYPE, &state_element_type)); in CompileImpl()
|
| D | image_resize_ops.cc | 601 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(output_dtype, &output_type_)); in ResizeBilinearGradOp()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 557 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType( in Compute()
|