- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for shape_add (0.15 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc
padding_value, CreateScalarConstValue<int8_t>(builder, loc, 0)); SmallVector<int64_t> shape_add(rank, 0); shape_add[pack_dim] = 1; shape_value = builder.create<TF::AddOp>( loc, shape_type, shape_value, CreateConstValue<int64_t>(builder, loc, {rank}, shape_add)); } packed_shape[pack_dim] /= 2; SmallVector<int64_t> divisor(rank, 1); divisor[pack_dim] = 2;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
std::vector<int64_t>* shape_arr = &tf_dlm_tensor_ctx->shape; std::vector<int64_t>* stride_arr = &tf_dlm_tensor_ctx->strides; shape_arr->resize(ndim); stride_arr->resize(ndim, 1); for (int i = 0; i < ndim; i++) { (*shape_arr)[i] = tensor->dim_size(i); } for (int i = ndim - 2; i >= 0; --i) { (*stride_arr)[i] = (*shape_arr)[i + 1] * (*stride_arr)[i + 1]; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 15 09:49:45 UTC 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/experimental/saved_model/internal/saved_model_api_test.cc
const TF_TensorSpec* tensor_spec_a = TF_SignatureDefParamTensorSpec(param_a); const TF_Shape* shape_a = TF_TensorSpecShape(tensor_spec_a); // Input "a" is a scalar, float32 tensor EXPECT_EQ("a", std::string(TF_SignatureDefParamName(param_a))); EXPECT_EQ(TF_FLOAT, TF_TensorSpecDataType(tensor_spec_a)); EXPECT_EQ(0, TF_ShapeDims(shape_a)); const TF_SignatureDefParam* param_b = TF_SignatureDefParamListGet(args, 1);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 23 08:08:45 UTC 2024 - 21.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// Shape op legalization. //===----------------------------------------------------------------------===// // ----- // CHECK-LABEL: func @shape_1D func.func @shape_1D(%arg0: tensor<?xf32>) -> tensor<1xi32> { // CHECK: [[SHAPE:%.+]] = shape.shape_of %arg0 // CHECK: [[TENSOR:%.+]] = arith.index_cast [[SHAPE]] : tensor<1xindex> to tensor<1xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0)