Searched refs:indices_type (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | softmax_op.cc | 205 const DataType indices_type = input_type(1); in Compile() local
224 xla::And(xla::Le(XlaHelpers::Zero(builder, indices_type), indices), in Compile()
226 builder, indices_type, depth))), in Compile()
|
| D | gather_op.cc | 204 DataType indices_type = context->input_type(1); in Compile() local
228 indices_type, builder, &gather)); in Compile()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | one_hot_test.cc | 34 T off_value = 0, TensorType indices_type = TensorType_INT32) { in OneHotOpModel() argument
35 indices_ = AddInput(indices_type); in OneHotOpModel()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 1606 auto indices_type = dynamic_slice->operand(1)->shape().element_type(); in OptimizeDotOfGather() local
1607 Shape s_shape = ShapeUtil::MakeShape(indices_type, {1}); in OptimizeDotOfGather()
1608 Shape d_shape = ShapeUtil::MakeShape(indices_type, {2}); in OptimizeDotOfGather()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1791 const tensorflow::DataType indices_type = in ConvertReduceOperator() local
1793 (*new_op->mutable_attr())["Tidx"].set_type(indices_type); in ConvertReduceOperator()
|