Searched refs:input_features (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/python/tpu/ |
| D | feature_column_test.py | 233 input_features = {'aaa': input_a, 'bbb': input_b}
275 fc._LazyBuilder(input_features))
277 fc._LazyBuilder(input_features))
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_pad_for_tensor_cores.cc | 166 auto input_features = in PadForTensorCores() local
170 if (input_features == 3 && (output_features == 32 || output_features == 64)) { in PadForTensorCores()
|
| D | cudnn_conv_rewriter.cc | 480 int64 input_features = new_shape.dimensions(input_feature_dimension); in MatchBackwardInput() local
483 input_features / conv->feature_group_count()); in MatchBackwardInput()
|
| /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/ |
| D | saved_model_export_utils_test.py | 507 input_features = constant_op.constant(["10"])
510 "features": input_features
577 input_features = constant_op.constant(["10"])
578 input_ops = ({"features": input_features}, None)
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | parse_example_dataset_test.py | 153 input_features = {
174 input_features,
181 input_features,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | parse_single_example_op_test.py | 165 input_features = {
187 "features": input_features,
196 "features": input_features,
|
| D | parsing_ops_test.py | 166 input_features = {
188 "features": input_features,
199 "features": input_features,
|
| /external/tensorflow/tensorflow/tools/docker/notebooks/ |
| D | 1_hello_tensorflow.ipynb | 530 " input_features = tf.constant(np.reshape([1, 0, 0, 1], (1, 4)).astype(np.float32))\n",
532 " output = tf.matmul(input_features, weights)\n",
534 " print(input_features.eval())\n",
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 1698 const int64 input_features = lhs.dimensions(dnums.input_feature_dimension()); in InferConvolveShape() local
1721 if (input_features % feature_group_count != 0 || in InferConvolveShape()
1722 input_features / feature_group_count != kernel_input_features) { in InferConvolveShape()
1729 input_features, feature_group_count, kernel_input_features, in InferConvolveShape()
|
| /external/tensorflow/tensorflow/python/feature_column/ |
| D | feature_column_test.py | 5744 input_features = {
5788 _LazyBuilder(input_features))
5790 _LazyBuilder(input_features))
5815 input_features = {'aaa': input_a, 'bbb': input_b}
5842 input_features, [embedding_column_a, embedding_column_b],
5871 input_features = {
5905 _LazyBuilder(input_features))
5907 _LazyBuilder(input_features))
|
| D | feature_column_v2_test.py | 7490 input_features = {'aaa': input_a, 'bbb': input_b}
7532 fc.FeatureTransformationCache(input_features), None)
7534 fc.FeatureTransformationCache(input_features), None)
7567 input_features = {
7602 fc.FeatureTransformationCache(input_features), None)
7604 fc.FeatureTransformationCache(input_features), None)
|