Searched defs:feature_shape (Results 1 – 5 of 5) sorted by relevance
366 shape_inference::ShapeHandle feature_shape; in __anon770f674c0a02() local418 shape_inference::ShapeHandle feature_shape; in __anon770f674c0b02() local462 shape_inference::ShapeHandle feature_shape; in __anon770f674c0c02() local491 shape_inference::ShapeHandle feature_shape; in __anon770f674c0d02() local533 shape_inference::ShapeHandle feature_shape; in __anon770f674c0f02() local762 ShapeHandle feature_shape; in __anon770f674c1502() local852 ShapeHandle feature_shape; in __anon770f674c1b02() 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