Searched refs:max_argmax (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | arithmetic.cc | 164 XlaOp max_argmax = Reduce(builder, {input, iota}, in ArgMinMax() local
167 XlaOp argmax = GetTupleElement(max_argmax, 1); in ArgMinMax()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_parser_test.cc | 1196 max_argmax { in CreateTestCases()
1212 …(f32[], s32[]) reduce(values, indices, init_value, init_index), dimensions={0}, to_apply=max_argmax in CreateTestCases()
|