Home
last modified time | relevance | path

Searched refs:w_shape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dconv_constants.cc251 const auto& w_shape = attr.weights.shape; in IsConvConstantsSupported() local
252 const int src_depth = DivideRoundUp(w_shape.i, 4); in IsConvConstantsSupported()
253 const int dst_depth = DivideRoundUp(w_shape.o, 4); in IsConvConstantsSupported()
255 use_dot_conv ? w_shape.o * src_depth * 4 : w_shape.i * dst_depth * 4; in IsConvConstantsSupported()
256 const int filters_count = aligned_ch_count * w_shape.h * w_shape.w; in IsConvConstantsSupported()
262 const int flt4_registers = DivideRoundUp(w_shape.o, 4); in IsConvConstantsSupported()
/external/tensorflow/tensorflow/python/keras/
Dconstraints.py223 w_shape = w.shape
224 if w_shape.rank is None or w_shape.rank != 4:
226 'The weight tensor must be of rank 4, but is of shape: %s' % w_shape)
228 height, width, channels, kernels = w_shape
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dself_adjoint_eig.cc119 TF_ASSIGN_OR_RETURN(Shape w_shape, builder->GetShape(jacobi_update.w)); in Update()
120 const std::vector<int64> batch_dims(w_shape.dimensions().begin(), in Update()
121 w_shape.dimensions().end() - 2); in Update()
122 const int64 num_dims = w_shape.rank(); in Update()
176 CreateScalarAddComputation(w_shape.element_type(), builder), in Update()
183 CreateScalarAddComputation(w_shape.element_type(), builder), in Update()
/external/libxaac/decoder/
Dixheaacd_ext_ch_ele.c415 WORD32 window, const WORD32 w_shape, in ixheaacd_estimate_dmx_im() argument
424 ixheaacd_mdst_fcoeff_longshort_curr[prev_w_shape][w_shape]; in ixheaacd_estimate_dmx_im()
428 mdst_fcoeff_curr = ixheaacd_mdst_fcoeff_start_curr[prev_w_shape][w_shape]; in ixheaacd_estimate_dmx_im()
432 mdst_fcoeff_curr = ixheaacd_mdst_fcoeff_stop_cur[prev_w_shape][w_shape]; in ixheaacd_estimate_dmx_im()
438 ixheaacd_mdst_fcoeff_stopstart_cur[prev_w_shape][w_shape]; in ixheaacd_estimate_dmx_im()
444 ixheaacd_mdst_fcoeff_stopstart_cur[prev_w_shape][w_shape]; in ixheaacd_estimate_dmx_im()
/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py1509 w_shape = constant_op.constant([2, 2, 2, 1, 1], shape=[5])
1510 dw = gen_nn_ops.conv3d_backprop_filter_v2(x, w_shape, dy, strides, 'SAME')
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc6058 auto w_shape = llvm::to_vector<4>(batch_dims); in ComputeWYRepresentation() local
6059 w_shape.append({m, n}); in ComputeWYRepresentation()
6061 RankedTensorType::get(w_shape, data_type), in ComputeWYRepresentation()
6062 w, GetI64ElementsAttr(w_shape, rewriter)); in ComputeWYRepresentation()