Searched defs:out_shapes (Results 1 – 3 of 3) sorted by relevance
34 const std::vector<std::vector<int32_t>>& out_shapes) { in run_view_test_cases()66 std::vector<std::vector<int32_t>> out_shapes = { in test_dtype() local85 std::vector<std::vector<int32_t>> out_shapes = { in test_empty_input() local
264 const std::vector<TensorShape>& out_shapes) { in BuildNodeDef()280 const std::vector<TensorShape>& out_shapes) { in MakeOp()
80 std::vector<shape_inference::ShapeHandle> out_shapes(in_shapes.size()); in __anoncf35e64b0102() local