Searched refs:rank_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_test.py | 832 rank_op = lambda x: array_ops.rank_internal(x, optimize=False) function
835 rank_op, dtype(7), expected=np.int32(0))
837 rank_op, np.array([[], []], dtype=dtype), expected=np.int32(2))
839 rank_op, np.array([-1, 1], dtype=dtype), expected=np.int32(1))
841 rank_op, np.array([[-1, 1]], dtype=dtype), expected=np.int32(2))
843 rank_op,
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 1311 tensorflow::NodeDef* rank_op = tensorflow_graph->add_node(); in ConvertRankOperator() local
1312 rank_op->set_op("Rank"); in ConvertRankOperator()
1313 rank_op->set_name(src_op.outputs[0]); in ConvertRankOperator()
1315 *rank_op->add_input() = src_op.inputs[0]; in ConvertRankOperator()
1316 (*rank_op->mutable_attr())["T"].set_type( in ConvertRankOperator()
|