Home
last modified time | relevance | path

Searched defs:constant_shape (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/tools/benchmark/
H A Dbenchmark_model_test.cc37 const TensorShape constant_shape({input_height, input_width}); in CreateTestGraph() local
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_runtime/
H A Dmodel.h355 const int64_t* constant_shape(int64_t idx) const { in constant_shape() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_layout_assignment_test.cc221 auto constant_shape = ShapeUtil::MakeTupleShape({other_shape, rhs_shape}); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ops/
H A Dquantization_ops_test.cc162 TensorShape constant_shape = TensorShape({channels}); in __anon9c93284d0202() local