Searched refs:TF_ATTR_SHAPE (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | tf_attrtype.h | 29 TF_ATTR_SHAPE = 5, enumerator
|
| D | c_api_test.cc | 2148 EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, -1); in TEST_F()
2159 EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, sz); in TEST_F()
2180 EXPECT_TF_META("v", list_size, TF_ATTR_SHAPE, total_ndims); in TEST_F()
2208 EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, 4); in TEST_F()
2241 EXPECT_TF_META("v", 2, TF_ATTR_SHAPE, 8); in TEST_F()
|
| D | c_api.cc | 1249 SINGLE_CASE(kShape, TF_ATTR_SHAPE, in TF_OperationGetAttrMetadata()
1275 shape, TF_ATTR_SHAPE, metadata.total_size = 0; in TF_OperationGetAttrMetadata()
1300 metadata.type = TF_ATTR_SHAPE; in TF_OperationGetAttrMetadata()
|
| /external/tensorflow/tensorflow/go/ |
| D | attrs.go | 152 case C.TF_ATTR_SHAPE:
227 case C.TF_ATTR_SHAPE:
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.cc | 108 t |= TF_ATTR_SHAPE; in AttrTypeMapForOp()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 90 elif attr_type == int(pywrap_tfe.TF_ATTR_SHAPE):
92 elif attr_type == [int(pywrap_tfe.TF_ATTR_SHAPE)]:
|
| D | backprop_test.py | 1053 actual = backprop.make_attr(int(pywrap_tfe.TF_ATTR_SHAPE), s)
1064 backprop.make_attr([int(pywrap_tfe.TF_ATTR_SHAPE)], shape_list))
|
| D | pywrap_tfe_src.cc | 404 } else if (type == TF_ATTR_SHAPE) { in SetOpAttrList()
569 } else if (type == TF_ATTR_SHAPE) { in SetOpAttrListDefault()
643 } else if (type == TF_ATTR_SHAPE) { in SetOpAttrScalar()
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 1081 .value("TF_ATTR_SHAPE", TF_ATTR_SHAPE) in PYBIND11_MODULE()
|