Searched defs:feature_shape (Results 1 – 5 of 5) sorted by relevance
366 shape_inference::ShapeHandle feature_shape; in __anon9348109b0a02() local418 shape_inference::ShapeHandle feature_shape; in __anon9348109b0b02() local462 shape_inference::ShapeHandle feature_shape; in __anon9348109b0c02() local491 shape_inference::ShapeHandle feature_shape; in __anon9348109b0d02() local533 shape_inference::ShapeHandle feature_shape; in __anon9348109b0f02() local762 ShapeHandle feature_shape; in __anon9348109b1502() local852 ShapeHandle feature_shape; in __anon9348109b1b02() local
26 def create_example_data(feature_dtype, feature_shape): argument
30 def create_example_data(feature_dtype, feature_shape): argument
178 const Shape feature_shape = scale->shape(); in HandleBatchNormTraining() local309 const Shape feature_shape = scale->shape(); in HandleBatchNormInference() local417 const Shape feature_shape = scale->shape(); in HandleBatchNormGrad() local
1570 Shape feature_shape = in InferBatchNormGradShape() local