Home
last modified time | relevance | path

Searched defs:out_shapes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/ops/
Dbatch_ops.cc80 std::vector<shape_inference::ShapeHandle> out_shapes(in_shapes.size()); in __anon1af940cb0102() local
/external/executorch/kernels/test/
Dop_view_copy_test.cpp34 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() local
85 std::vector<std::vector<int32_t>> out_shapes = { in test_empty_input() local
/external/tensorflow/tensorflow/core/kernels/
Dscoped_allocator_ops_test.cc264 const std::vector<TensorShape>& out_shapes) { in BuildNodeDef()
280 const std::vector<TensorShape>& out_shapes) { in MakeOp()