| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| xla_context.cc | 83 xla::PrimitiveType xla_type; local
84 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type));
86 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x");
88 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y");
99 xla::PrimitiveType xla_type; local
100 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type));
102 xla::Parameter(&b, 0, xla::ShapeUtil::MakeShape(xla_type, {}), "x");
104 xla::Parameter(&b, 1, xla::ShapeUtil::MakeShape(xla_type, {}), "y");
115 xla::PrimitiveType xla_type;
116 TF_CHECK_OK(DataTypeToPrimitiveType(type, &xla_type));
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| quantize_and_dequantize_op.cc | 56 xla::PrimitiveType xla_type; variable
57 OP_REQUIRES_OK(ctx, DataTypeToPrimitiveType(data_type, &xla_type));
70 min_range = ReduceAll(input, xla::MaxValue(b, xla_type), *fmin);
71 max_range = ReduceAll(input, xla::MinValue(b, xla_type), *fmax);
98 xla_type));
101 xla_type)) -
105 max_quantized = Pow(two, ConvertElementType(num_bits, xla_type)) - one;
113 xla::MaxFiniteValue(b, xla_type));
116 xla::MaxFiniteValue(b, xla_type));
|
| softmax_op.cc | 52 const xla::PrimitiveType xla_type = ctx->input_xla_type(0); variable
60 xla::Reduce(logits, xla::MinValue(b, xla_type), max_func, {kClassDim});
92 XlaOpKernelContext* ctx, DataType type, xla::PrimitiveType xla_type,
102 xla::Reduce(logits, xla::MinValue(b, xla_type), max_func, {kClassDim});
161 const xla::PrimitiveType xla_type = ctx->input_xla_type(0); variable
167 CrossEntropyWithLogits(ctx, type, xla_type, logits, labels);
|
