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()
|