Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dc_api_experimental.cc545 void TF_ShapeAndTypeListSetShape(TF_ShapeAndTypeList* shape_list, int index, in TF_ShapeAndTypeListSetShape()
556 void TF_ShapeAndTypeListSetUnknownShape(TF_ShapeAndTypeList* shape_list, in TF_ShapeAndTypeListSetUnknownShape()
565 void TF_ShapeAndTypeListSetDtype(TF_ShapeAndTypeList* shape_list, int index, in TF_ShapeAndTypeListSetDtype()
572 void TF_DeleteShapeAndTypeList(TF_ShapeAndTypeList* shape_list) { in TF_DeleteShapeAndTypeList()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_tf_dialect_op.cc73 auto& shape_list = *value.mutable_list(); in SetShapeAttribute() local
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_test.py402 def shape_list(tensor): function
/external/tensorflow/tensorflow/python/training/
Dinput.py670 def _merge_shapes(shape_list, enqueue_many): argument
/external/tensorflow/tensorflow/core/framework/
Dop_kernel_test.cc813 std::vector<TensorShape> shape_list; member in tensorflow::__anonc25ee1c30111::GetAttrKernel
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding_test.cc4216 const auto& shape_list = node.attr().at("_output_shapes").list(); in TEST_F() local