Searched refs:xla_type (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_context.cc | 83 xla::PrimitiveType xla_type; in GetOrCreateMax() local
84 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateMax()
86 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x"); in GetOrCreateMax()
88 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y"); in GetOrCreateMax()
99 xla::PrimitiveType xla_type; in GetOrCreateMin() local
100 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateMin()
102 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x"); in GetOrCreateMin()
104 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y"); in GetOrCreateMin()
115 xla::PrimitiveType xla_type; in GetOrCreateAdd() local
116 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type)); in GetOrCreateAdd()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | quantize_and_dequantize_op.cc | 56 xla::PrimitiveType xla_type; in Compile() local
57 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(data_type, &xla_type)); in Compile()
70 min_range = ReduceAll(input, xla::MaxValue(b, xla_type), *fmin); in Compile()
71 max_range = ReduceAll(input, xla::MinValue(b, xla_type), *fmax); in Compile()
98 xla_type)); in Compile()
101 xla_type)) - in Compile()
105 max_quantized = Pow(two, ConvertElementType(num_bits, xla_type)) - one; in Compile()
113 xla::MaxFiniteValue(b, xla_type)); in Compile()
116 xla::MaxFiniteValue(b, xla_type)); in Compile()
|
| D | softmax_op.cc | 52 const xla::PrimitiveType xla_type = ctx->input_xla_type(0); in Compile() local
60 xla::Reduce(logits, xla::MinValue(b, xla_type), max_func, {kClassDim}); in Compile()
92 XlaOpKernelContext* ctx, DataType type, xla::PrimitiveType xla_type, in CrossEntropyWithLogits() argument
102 xla::Reduce(logits, xla::MinValue(b, xla_type), max_func, {kClassDim}); in CrossEntropyWithLogits()
161 const xla::PrimitiveType xla_type = ctx->input_xla_type(0); in Compile() local
167 CrossEntropyWithLogits(ctx, type, xla_type, logits, labels); in Compile()
|
| D | resampler_ops.cc | 54 xla::PrimitiveType xla_type) { in BilinearWeights() argument
57 first_term = xla::ConvertElementType(first_term, xla_type); in BilinearWeights()
89 sign_change = xla::ConvertElementType(sign_change, xla_type); in BilinearWeights()
102 flipped, xla::One(ctx->builder(), xla_type), in BilinearWeights()
103 xla::CreateScalarMultiplyComputation(xla_type, ctx->builder()), in BilinearWeights()
169 xla::PrimitiveType xla_type) { in ScatterToGradData() argument
188 xla::CreateScalarAddComputation(xla_type, ctx->builder()), in ScatterToGradData()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_state_ops.h | 556 xla::PrimitiveType xla_type; in Compute()
558 ctx->expected_output_dtype(output), &xla_type)); in Compute()
559 if (xla_type != subshape->element_type()) { in Compute()
|