| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | cast_op.cc | 39 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(src_dtype_, &src_type_)); in CastOp()
40 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dst_dtype_, &dst_type_)); in CastOp()
110 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(src_dtype_, &src_type_)); in BitcastOp()
111 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dst_dtype_, &dst_type_)); in BitcastOp()
|
| D | categorical_op.cc | 77 DataTypeToPrimitiveType(input_type(0), &uniform_xla_type)); in Compile()
89 DataTypeToPrimitiveType(input_type(0), &uniform_xla_type)); in Compile()
95 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_type(0), &type)); in Compile()
113 DataTypeToPrimitiveType(output_type(0), &xla_output_type)); in Compile()
|
| D | reduction_ops_common.cc | 35 ctx, DataTypeToPrimitiveType(reduction_type_, &xla_reduction_type_)); in XlaReductionOp()
114 TF_CHECK_OK(DataTypeToPrimitiveType(reduction_type_, &type)); in Compile()
|
| D | empty_op.cc | 35 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype_, &type_)); in EmptyOp()
|
| D | batch_matmul_op.cc | 38 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(output_type, &xla_type)); in BatchMatMulOp()
|
| D | xla_dot_op.cc | 78 OP_REQUIRES_OK(context, DataTypeToPrimitiveType(preferred_element_dtype, in XlaDotV2Op()
|
| D | pooling_ops.cc | 79 ctx, DataTypeToPrimitiveType(reduction_type_, &xla_reduction_type_)); in PoolingOp()
402 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_type(2), &element_type)); in Compile()
495 ctx, DataTypeToPrimitiveType(reduction_type, &xla_reduction_type)); in Compile()
509 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(ctx->input_type(1), in Compile()
688 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_type(2), &element_type)); in Compile()
|
| D | index_ops.cc | 63 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(index_type, &index_xla_type)); in Compile()
|
| D | batch_norm_op.cc | 81 DataTypeToPrimitiveType(ctx->input_type(0), &input_type)); in CompileImpl()
84 DataTypeToPrimitiveType(ctx->input_type(1), &scale_type)); in CompileImpl()
|
| D | in_topk_op.cc | 38 DataTypeToPrimitiveType(targets_dtype_, &targets_type_)); in InTopKOp()
|
| D | sharding_util_ops.cc | 142 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(input_dtype, &type)); in CompileInternal()
314 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(dtype_, &type)); in CompileInternal()
|
| D | segment_reduction_ops.cc | 33 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype, &type_)); in UnsortedSegmentReduce()
|
| D | extract_image_patches_op.cc | 95 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(ctx->input_type(0), &type)); in Compile()
|
| D | stateful_random_ops.cc | 210 DataTypeToPrimitiveType(STATE_ELEMENT_DTYPE, &state_element_type)); in CompileImpl()
469 ctx, DataTypeToPrimitiveType(STATE_ELEMENT_DTYPE, &state_element_type)); in Compile()
|
| D | quantize_and_dequantize_op.cc | 62 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(data_type, &xla_type)); in Compile()
|
| D | tensor_list_ops.cc | 168 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype_, &type)); in Compile()
242 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype_, &type)); in Compile()
|
| D | unique_op.cc | 53 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype, &idx_type_)); in UniqueOpBase()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.cc | 91 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateMax()
107 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateMin()
123 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateAdd()
139 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateMul()
|
| D | xla_helpers.cc | 45 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in Zero()
51 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in One()
58 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in IntegerLiteral()
65 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in FloatLiteral()
133 TF_CHECK_OK(DataTypeToPrimitiveType(new_element_type, &convert_to)); in ConvertElementType()
|
| D | type_util.h | 27 Status DataTypeToPrimitiveType(DataType data_type, xla::PrimitiveType* type);
|
| D | shape_util.cc | 107 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(dtype, &type)); in TensorShapeToXLAShape()
140 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(dtype, &type)); in TensorShapeToXLAShape()
|
| D | type_util.cc | 24 Status DataTypeToPrimitiveType(DataType data_type, xla::PrimitiveType* type) { in DataTypeToPrimitiveType() function
|
| D | literal_util.cc | 96 DataTypeToPrimitiveType(host_tensor->dtype(), &primitive_type)); in CopyLiteralToHostTensor()
|
| D | xla_op_kernel.cc | 121 Status status = DataTypeToPrimitiveType(input_type(index), &type); in input_xla_type()
131 Status status = DataTypeToPrimitiveType(InputType(name), &type); in InputXlaType()
661 Status status = DataTypeToPrimitiveType(expected_output_dtype(index), &type); in output_xla_type()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 585 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType( in Compute()
|