Searched refs:tf_shape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/saved_model/internal/ |
| D | BUILD | 325 "//tensorflow/c:tf_shape",
356 "//tensorflow/c:tf_shape",
|
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 758 const TensorShape& tf_shape,
763 tf_shape, output));
781 TensorShape tf_shape; in AllocTmpBuffer() local
783 tf_shape.AddDim(pd.get_size() / sizeof(T) + 1); in AllocTmpBuffer()
785 tf_shape, tensor_out)); in AllocTmpBuffer()
791 TensorShape tf_shape) { in AllocTmpBuffer() argument
793 tf_shape, tensor_out)); in AllocTmpBuffer()
|
| /external/tensorflow/tensorflow/c/ |
| D | BUILD | 270 name = "tf_shape",
271 srcs = ["tf_shape.cc"],
272 hdrs = ["tf_shape.h"],
|
| /external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
| D | normalization_test.py | 1121 tf_shape = [None, 5, 4]
1122 inp = array_ops.placeholder(dtypes.float32, tf_shape)
|