Searched defs:feature_shape (Results 1 – 5 of 5) sorted by relevance
366 shape_inference::ShapeHandle feature_shape; in __anon8a3fd5e50a02() local418 shape_inference::ShapeHandle feature_shape; in __anon8a3fd5e50b02() local462 shape_inference::ShapeHandle feature_shape; in __anon8a3fd5e50c02() local491 shape_inference::ShapeHandle feature_shape; in __anon8a3fd5e50d02() local533 shape_inference::ShapeHandle feature_shape; in __anon8a3fd5e50f02() local762 ShapeHandle feature_shape; in __anon8a3fd5e51502() local852 ShapeHandle feature_shape; in __anon8a3fd5e51b02() local
179 const Shape feature_shape = scale->shape(); in HandleBatchNormTraining() local308 const Shape feature_shape = scale->shape(); in HandleBatchNormInference() local414 const Shape feature_shape = scale->shape(); in HandleBatchNormGrad() local
1571 Shape feature_shape = in InferBatchNormGradShape() local
100 def __init__(self, batch_size, feature_shape): argument
1558 const auto feature_shape = feature_shape_t->vec<int32>(); in Compute() local