HomeSort by relevance Sort by last modified time
    Searched refs:sparse_shape (Results 1 - 6 of 6) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_hlo_support_checker_test.cc 53 const Shape sparse_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {10}, 2); local
55 HloInstruction::CreateParameter(0, sparse_shape, "param0"));
57 HloInstruction::CreateParameter(1, sparse_shape, "param1"));
59 sparse_shape, HloOpcode::kAdd, param0, param1));
68 HasSubstr(ShapeUtil::HumanStringWithLayout(sparse_shape)));
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_hlo_support_checker_test.cc 53 const Shape sparse_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {10}, 2); local
55 HloInstruction::CreateParameter(0, sparse_shape, "param0"));
57 HloInstruction::CreateParameter(1, sparse_shape, "param1"));
59 sparse_shape, HloOpcode::kAdd, param0, param1));
68 HasSubstr(ShapeUtil::HumanStringWithLayout(sparse_shape)));
  /external/tensorflow/tensorflow/contrib/tensor_forest/kernels/v4/
input_data.cc 107 const Tensor& sparse_shape) {
116 sparse_batch_size_ = sparse_shape.tensor<int64, 1>()(0);
input_data.h 71 const Tensor& sparse_shape);
  /external/tensorflow/tensorflow/contrib/tensor_forest/python/
tensor_forest.py 603 sparse_shape = []
607 sparse_shape = sparse_features.dense_shape
617 sparse_shape,
634 sparse_shape,
663 sparse_shape = []
667 sparse_shape = sparse_features.dense_shape
676 sparse_shape,
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 458 milliseconds