Home
last modified time | relevance | path

Searched defs:shape_list (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_experimental.cc552 void TF_ShapeAndTypeListSetShape(TF_ShapeAndTypeList* shape_list, int index, in TF_ShapeAndTypeListSetShape()
563 void TF_ShapeAndTypeListSetUnknownShape(TF_ShapeAndTypeList* shape_list, in TF_ShapeAndTypeListSetUnknownShape()
572 void TF_ShapeAndTypeListSetDtype(TF_ShapeAndTypeList* shape_list, int index, in TF_ShapeAndTypeListSetDtype()
579 void TF_DeleteShapeAndTypeList(TF_ShapeAndTypeList* shape_list) { in TF_DeleteShapeAndTypeList()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dexport_tf_dialect_op.cc73 auto& shape_list = *value.mutable_list(); in SetShapeAttribute() local
/aosp_15_r20/external/tensorflow/tensorflow/python/training/
H A Dinput.py665 def _merge_shapes(shape_list, enqueue_many): argument
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dop_kernel_test.cc697 std::vector<TensorShape> shape_list; member in tensorflow::__anondc0872320111::GetAttrKernel
/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dtpu_strategy_test.py539 def shape_list(tensor): function
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dregister_prim_ops.cpp2699 at::IntArrayRef shape_list( in __anonbfe5918f9402() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dconstant_folding_test.cc4221 const auto& shape_list = node.attr().at("_output_shapes").list(); in TEST_F() local