Searched refs:CreateScalarGtComputation (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | sorting.cc | 109 {input_and_iota}, CreateScalarGtComputation({S32}, builder), last_dim, in TopK()
136 CreateScalarGtComputation({input_shape.element_type(), S32}, in TopK()
199 CreateScalarGtComputation({input_shape.element_type(), S32}, in TopKWithPartitions()
230 CreateScalarGtComputation({input_shape.element_type(), S32}, in TopKWithPartitions()
|
| D | comparators.h | 42 XlaComputation CreateScalarGtComputation(
|
| D | comparators.cc | 121 XlaComputation CreateScalarGtComputation( in CreateScalarGtComputation() function
|
| D | comparators_test.cc | 52 : CreateScalarGtComputation({type}, test->builder()); in BuildComparatorAndComparisons()
|
| D | svd.cc | 763 CreateScalarGtComputation( in SortBySingularValuesAndPostProcessing()
769 CreateScalarGtComputation( in SortBySingularValuesAndPostProcessing()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | where_op.cc | 176 to_sort, xla::CreateScalarGtComputation(types_to_sort, ctx->builder()), in CompileWhereWithSort()
|
| D | image_ops.cc | 476 xla::CreateScalarGtComputation( in ComputeResult()
484 xla::CreateScalarGtComputation({scores_xla_type, xla::S32}, builder)); in ComputeResult()
|
| D | unique_op.cc | 228 xla::CreateScalarGtComputation({xla::S32, xla::S32}, ctx->builder()), 0, in CompileWithAxis()
|