Searched refs:TF_ATTR_SHAPE (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/go/ |
| D | attrs.go | 152 case C.TF_ATTR_SHAPE:
227 case C.TF_ATTR_SHAPE:
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 79 elif attr_type == pywrap_tensorflow.TF_ATTR_SHAPE:
81 elif attr_type == [pywrap_tensorflow.TF_ATTR_SHAPE]:
|
| D | backprop_test.py | 963 actual = backprop.make_attr(pywrap_tensorflow.TF_ATTR_SHAPE, s)
974 backprop.make_attr([pywrap_tensorflow.TF_ATTR_SHAPE], shape_list))
|
| D | pywrap_tfe_src.cc | 314 } else if (type == TF_ATTR_SHAPE) { in SetOpAttrList()
479 } else if (type == TF_ATTR_SHAPE) { in SetOpAttrListDefault()
553 } else if (type == TF_ATTR_SHAPE) { in SetOpAttrScalar()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.cc | 107 t |= TF_ATTR_SHAPE; in AttrTypeMapForOp()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_test.cc | 2147 EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, -1); in TEST_F()
2158 EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, sz); in TEST_F()
2179 EXPECT_TF_META("v", list_size, TF_ATTR_SHAPE, total_ndims); in TEST_F()
2207 EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, 4); in TEST_F()
2240 EXPECT_TF_META("v", 2, TF_ATTR_SHAPE, 8); in TEST_F()
|
| D | c_api.h | 696 TF_ATTR_SHAPE = 5, enumerator
|
| D | c_api.cc | 1662 SINGLE_CASE(kShape, TF_ATTR_SHAPE, in TF_OperationGetAttrMetadata()
1688 shape, TF_ATTR_SHAPE, metadata.total_size = 0; in TF_OperationGetAttrMetadata()
1713 metadata.type = TF_ATTR_SHAPE; in TF_OperationGetAttrMetadata()
|