| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | categorical_op.cc | 74 DataTypeToPrimitiveType(input_type(0), &uniform_xla_type)); in Compile()
86 DataTypeToPrimitiveType(input_type(0), &uniform_xla_type)); in Compile()
92 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_type(0), &type)); in Compile()
112 DataTypeToPrimitiveType(output_type(0), &xla_output_type)); in Compile()
|
| 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 | reduction_ops_common.cc | 35 ctx, DataTypeToPrimitiveType(reduction_type_, &xla_reduction_type_)); in XlaReductionOp()
112 TF_CHECK_OK(DataTypeToPrimitiveType(reduction_type_, &type)); in Compile()
|
| D | empty_op.cc | 35 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype_, &type_)); in EmptyOp()
|
| D | pooling_ops.cc | 71 ctx, DataTypeToPrimitiveType(reduction_type_, &xla_reduction_type_)); in PoolingOp()
354 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_type(2), &element_type)); in Compile()
447 ctx, DataTypeToPrimitiveType(reduction_type, &xla_reduction_type)); in Compile()
461 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(ctx->input_type(1), in Compile()
636 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(input_type(2), &element_type)); in Compile()
|
| D | index_ops.cc | 65 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(index_type, &index_xla_type)); in Compile()
|
| D | batch_norm_op.cc | 77 DataTypeToPrimitiveType(ctx->input_type(0), &input_type)); in CompileImpl()
80 DataTypeToPrimitiveType(ctx->input_type(1), &scale_type)); in CompileImpl()
|
| D | in_topk_op.cc | 38 DataTypeToPrimitiveType(targets_dtype_, &targets_type_)); in InTopKOp()
|
| D | sort_ops.cc | 80 OP_REQUIRES_OK(context, DataTypeToPrimitiveType(input_types_[i], in Compile()
|
| D | segment_reduction_ops.cc | 32 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 | quantize_and_dequantize_op.cc | 62 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(data_type, &xla_type)); in Compile()
|
| D | stateful_random_ops.cc | 209 DataTypeToPrimitiveType(STATE_ELEMENT_DTYPE, &state_element_type)); in CompileImpl()
468 ctx, DataTypeToPrimitiveType(STATE_ELEMENT_DTYPE, &state_element_type)); in Compile()
|
| D | tensor_list_ops.cc | 164 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype_, &type)); in Compile()
236 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(dtype_, &type)); in Compile()
|
| D | softmax_op.cc | 92 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(accumulation_type, in Compile()
|
| D | image_resize_ops.cc | 637 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(output_dtype, &output_type_)); in ResizeBilinearGradOp()
|
| /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 | 37 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in Zero()
43 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in One()
50 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in IntegerLiteral()
57 TF_CHECK_OK(DataTypeToPrimitiveType(data_type, &type)); in FloatLiteral()
125 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()
148 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 | 112 Status status = DataTypeToPrimitiveType(input_type(index), &type); in input_xla_type()
122 Status status = DataTypeToPrimitiveType(InputType(name), &type); in InputXlaType()
580 Status status = DataTypeToPrimitiveType(expected_output_dtype(index), &type); in output_xla_type()
|
| D | xla_compiler_test.cc | 1277 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(type, &ptype)); in TEST_F()
1348 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType(type, &ptype)); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 582 TF_RETURN_IF_ERROR(DataTypeToPrimitiveType( in Compute()
|