Searched refs:rank_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_test.py | 979 rank_op = lambda x: array_ops.rank_internal(x, optimize=False) function
982 rank_op, dtype(7), expected=np.int32(0))
984 rank_op, np.array([[], []], dtype=dtype), expected=np.int32(2))
986 rank_op, np.array([-1, 1], dtype=dtype), expected=np.int32(1))
988 rank_op, np.array([[-1, 1]], dtype=dtype), expected=np.int32(2))
990 rank_op,
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1324 tensorflow::NodeDef* rank_op = tensorflow_graph->add_node(); in ConvertRankOperator() local
1325 rank_op->set_op("Rank"); in ConvertRankOperator()
1326 rank_op->set_name(src_op.outputs[0]); in ConvertRankOperator()
1328 *rank_op->add_input() = src_op.inputs[0]; in ConvertRankOperator()
1329 (*rank_op->mutable_attr())["T"].set_type( in ConvertRankOperator()
|