Searched refs:TF_ATTR_SHAPE (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | tf_attrtype.h | 29 TF_ATTR_SHAPE = 5, enumerator
|
| D | kernels.cc | 300 SINGLE_CASE(kShape, TF_ATTR_SHAPE, in TF_OpKernelConstruction_GetAttrSize()
324 shape, TF_ATTR_SHAPE, *total_size = 0; in TF_OpKernelConstruction_GetAttrSize()
|
| D | c_api_test.cc | 2185 EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, -1); in TEST_F()
2196 EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, sz); in TEST_F()
2217 EXPECT_TF_META("v", list_size, TF_ATTR_SHAPE, total_ndims); in TEST_F()
2245 EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, 4); in TEST_F()
2278 EXPECT_TF_META("v", 2, TF_ATTR_SHAPE, 8); in TEST_F()
|
| D | c_api.cc | 1251 SINGLE_CASE(kShape, TF_ATTR_SHAPE, in TF_OperationGetAttrMetadata()
1277 shape, TF_ATTR_SHAPE, metadata.total_size = 0; in TF_OperationGetAttrMetadata()
1302 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 | 119 t |= TF_ATTR_SHAPE; in AttrTypeMapForOp()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | backprop.py | 94 if attr_type == int(pywrap_tfe.TF_ATTR_SHAPE):
96 if attr_type == [int(pywrap_tfe.TF_ATTR_SHAPE)]:
|
| D | pywrap_tfe_src.cc | 423 } else if (type == TF_ATTR_SHAPE) { in SetOpAttrList()
588 } else if (type == TF_ATTR_SHAPE) { in SetOpAttrListDefault()
662 } else if (type == TF_ATTR_SHAPE) { in SetOpAttrScalar()
|
| D | backprop_test.py | 1113 actual = backprop.make_attr(int(pywrap_tfe.TF_ATTR_SHAPE), s)
1124 backprop.make_attr([int(pywrap_tfe.TF_ATTR_SHAPE)], shape_list))
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 1507 .value("TF_ATTR_SHAPE", TF_ATTR_SHAPE) in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 1685 ASSERT_EQ(TF_ATTR_SHAPE, in TEST()
|