Searched refs:GetBroadcastSize (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | ops_util.cc | 40 Status GetBroadcastSize(const int index, const int in_size, const int ksize, in GetBroadcastSize() function
|
| D | ops_util.h | 38 Status GetBroadcastSize(const int index, const int in_size, const int ksize,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 230 GetBroadcastSize(p, input_size[0], window[0], stride[0], in launch()
235 GetBroadcastSize(r, input_size[1], window[1], stride[1], in launch()
240 context, GetBroadcastSize(c, input_size[2], window[2], stride[2], in launch()
405 GetBroadcastSize(p, input_size[0], window[0], stride[0], in launch()
410 GetBroadcastSize(r, input_size[1], window[1], stride[1], in launch()
415 context, GetBroadcastSize(c, input_size[2], window[2], stride[2], in launch()
|
| D | ops_util_test.cc | 126 Status status = GetBroadcastSize( in VerifyBoundaries()
135 GetBroadcastSize(bcast.input.index, bcast.input.in_size, in VerifyBcastValues()
|
| D | avgpooling_op.cc | 310 GetBroadcastSize(r, in_rows, window_rows, row_stride, in Compute()
320 GetBroadcastSize(c, in_cols, window_cols, col_stride, in Compute()
|