Searched refs:shape_func (Results 1 – 6 of 6) sorted by relevance
98 shape_func=lambda op: [op.inputs[0].shape],107 shape_func=lambda op: [op.inputs[0].shape],
344 @function.Defun(shape_func=_swish_shape, func_name="swish_grad", noinline=True)361 shape_func=_swish_shape,
767 shape_func=_ctc_loss_shape)
213 shape_func=None, argument254 self._shape_func = shape_func
878 dtypes.float32, shape_func=lambda op: [op.inputs[0].get_shape()])883 shape_func=lambda op: [[1] + op.inputs[0].get_shape().as_list()])
2630 python_grad_func=_scale_gradient_grad, shape_func=_scale_gradient_shape)