Searched refs:w_shape (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
D | conv_constants.cc | 251 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/ |
D | constraints.py | 223 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/ |
D | self_adjoint_eig.cc | 119 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/ |
D | ixheaacd_ext_ch_ele.c | 415 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/ |
D | layout_optimizer_test.py | 1509 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/ |
D | legalize_tf.cc | 6058 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()
|