Home
last modified time | relevance | path

Searched refs:input_features (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_v2_test.py391 input_features = {'aaa': input_a, 'bbb': input_b}
438 embedding_lookup_a = dense_features(input_features)
439 embedding_lookup_b = sequence_features(input_features)
498 input_features = {'inp': input_sparse_tensor}
521 dense_features(input_features)
543 dense_features(input_features)
582 input_features = {'inp': input_sparse_tensor}
631 embedding_lookup = dense_features(input_features)
649 embedding_lookup = dense_features(input_features)
678 input_features = {'inp': input_sparse_tensor}
[all …]
Dfeature_column_test.py254 input_features = {'aaa': input_a, 'bbb': input_b}
296 fc._LazyBuilder(input_features))
298 fc._LazyBuilder(input_features))
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc228 auto input_features = in TryResolvePaddedShapesForTensorCore() local
232 if (input_features == 3 && (output_features == 32 || output_features == 64)) { in TryResolvePaddedShapesForTensorCore()
Dgpu_conv_rewriter.cc510 int64_t input_features = rhs->shape().dimensions(input_feature_dimension); in MatchBackwardInput() local
517 CHECK_EQ(input_features % num_groups, 0) in MatchBackwardInput()
/external/tensorflow/tensorflow/python/kernel_tests/io_ops/
Dparse_single_example_op_test.py161 input_features = {
183 "features": input_features,
192 "features": input_features,
Dparsing_ops_test.py154 input_features = {
177 "features": input_features,
188 "features": input_features,
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dparse_example_dataset_test.py151 input_features = {
172 input_features,
179 input_features,
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dshape_inference.cc1664 auto input_features = input_tensor_shape[dnums.input_feature_dimension()]; in PrecheckForXlaConvV2Op() local
1694 if (input_features % feature_group_count_val != 0 || in PrecheckForXlaConvV2Op()
1695 input_features / feature_group_count_val != kernel_input_features) { in PrecheckForXlaConvV2Op()
1702 << input_features << ") in lhs.\n"; in PrecheckForXlaConvV2Op()
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_v2_test.py5749 input_features = {'aaa': input_a, 'bbb': input_b}
5810 fc.FeatureTransformationCache(input_features), None)
5812 fc.FeatureTransformationCache(input_features), None)
5863 input_features = {'aaa': input_a, 'bbb': input_b}
5906 fc.FeatureTransformationCache(input_features), None)
5908 fc.FeatureTransformationCache(input_features), None)
5958 input_features = {
5994 fc.FeatureTransformationCache(input_features), None)
5996 fc.FeatureTransformationCache(input_features), None)
Dfeature_column_test.py5774 input_features = {'aaa': input_a, 'bbb': input_b}
5836 _LazyBuilder(input_features))
5838 _LazyBuilder(input_features))
5879 input_features = {'aaa': input_a, 'bbb': input_b}
5906 input_features, [embedding_column_a, embedding_column_b],
5937 input_features = {
5971 _LazyBuilder(input_features))
5973 _LazyBuilder(input_features))
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc1739 const int64_t input_features = in InferConvolveShape() local
1761 if (input_features % feature_group_count != 0 || in InferConvolveShape()
1762 input_features / feature_group_count != kernel_input_features) { in InferConvolveShape()
1769 input_features, feature_group_count, kernel_input_features, in InferConvolveShape()